如何乘以两个4D TensorVariable,python

时间:2016-12-17 05:06:43

标签: python tensorflow theano

我有两个4D TensorVariable,X=[100,1,28,28]W=[100,1,28,28]X包含100张图片(单通道),大小为(28,28)W100图片的相应权重。我们希望将每个图像乘以其权重。换句话说,我想要100图像,其中每个单独的图像来自T.dot(a,b),其中a是一个图像,b是其权重。

T.dot(X,W)对我不起作用,因为它给了我一个5D TensorVariable。

1 个答案:

答案 0 :(得分:0)

Theano T.dot函数是矩阵乘法(内积)。你想要一个逐元素的乘法,正如documentation中所描述的那样,只需使用*运算符完成,即

X * W