Keras中的高斯密度函数实现

Question

我正在尝试在keras（python）中实现高斯密度函数求值函数。我尝试了以下方法，但它无法正常工作

def normal(z, mean, var, z_dim):
    return 1/(K.pow(2 * np.pi, z_dim / 2) * K.prod(var, 1)) * K.exp(- 1 / 2 * K.sum(K.square(z - mean) / var, 1))

我收到以下错误。

 Output tensors to a Model must be Keras tensors. Found: Tensor("mul_2207:0", shape=(?,), dtype=float32)