在TensorFlow中是否有与Matlab中相同的bsxfun?

时间:2016-06-19 06:17:52

标签: machine-learning neural-network tensorflow conv-neural-network

我试图将以下MATLAB代码转换为tensorflow:

WW = sum(W.^2, 1); % ( 1 x D^(l)= sum( (D^(l-1) x D^(l)), 1 )
XX = sum(A.^2, 2); % (M x 1) = sum( (M x D^(l-1)), 2 )
bsxfun(@plus, WW, XX) ; % (M x D^(l)) - (M x D^(l)) = (M x D^(l-1)) * (D^(l-1) x D^(l)) - (M x D^(l))

这是非常简单的MATLAB代码,并且想知道TensorFlow中是否存在等效代码。理想情况下,W和/或X应该是tf.Variable(init)变量,因为我想根据每个变量计算导数。

1 个答案:

答案 0 :(得分:2)

Tensorflow和NumPy一样做广播。

你可以做到

WW + XX

并且它会自己弄清楚尺寸

请参阅文档here