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