标签: python tensorflow theano
我有两个4D TensorVariable,X=[100,1,28,28]和W=[100,1,28,28]。 X包含100张图片(单通道),大小为(28,28)。 W是100图片的相应权重。我们希望将每个图像乘以其权重。换句话说,我想要100图像,其中每个单独的图像来自T.dot(a,b),其中a是一个图像,b是其权重。
X=[100,1,28,28]
W=[100,1,28,28]
X
100
(28,28)
W
T.dot(a,b)
a
b
T.dot(X,W)对我不起作用,因为它给了我一个5D TensorVariable。
T.dot(X,W)
答案 0 :(得分:0)
Theano T.dot函数是矩阵乘法(内积)。你想要一个逐元素的乘法,正如documentation中所描述的那样,只需使用*运算符完成,即
T.dot
*
X * W