machine-learning - TensorFlow 中是否有与 Matlab 中等效的 bsxfun ？

Question

我试图将以下 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) 变量，因为我想计算每个变量的导数变量。

Answer 1

Tensorflow 与 NumPy 一样，可以进行广播。

你可以做到

WW + XX

它会自己计算出尺寸

请参阅文档 here