TensorFlow: Оценка функции через тензор

Question

Sci-Py позволяет мне выполнять следующие функции: norm.pdf

x = 1; 
mu = [1,2,3,4]; 
sigma = [1,1,1,1]; 
scipy.stats.norm.pdf(x,mu,sigma) 

Это в основном даст мне массив, где каждый элемент соответствует плотности вероятности х учитывая соответствующее среднее и дисперсию.

Я пытаюсь сделать это с той разницей, что x, mu и var теперь являются тензорами (x - ранг-0, mu и var - ранга 1). Я хочу, чтобы norm.pdf также был тензором ранга 1. Я пытаюсь сделать это на TensorFlow, но не знаю, как это сделать.

Предложения?

Answer 1

Использование TensorFlow, вы можете сделать это в 2 этапа с tf.contrib.distributions.Normal и tf.contrib.distributions.Distribution.pdf

# Define a batch of 4 scalar valued Normals (mu=means, sigma=standard deviations) 
dist = tf.contrib.distributions.Normal(mu=[1,2,3,4], sigma=[1,1,1,1]) 

# Evaluate the pdf of the first distribution on 1 
# returning a length two tensor. 
dist.pdf(1)