Tensorflow：行列の各要素の操作方法

Question

2つの行列A（Nx100）とB（Mx100）があります。行列Bの各要素を持つ行列Aの各要素に対して演算を実行したい（ユークリッド距離を計算する）。通常のnumpyでは2つのforループを使って行う。

今のところマップ関数tf.map_fnを使って、行列Aの各要素を、操作する場所にマップします。A[i] - BここでA [i]はベクトルです。これは、TensorFlowがベクトルを行列の形にブロードキャストするために機能します。

これをもっと簡単に行う方法があるかどうかは疑問でした。何かのように：tf.do_magic(A, B, operation)？

ありがとうございます！ AとBの

Answer 1

ユークリッド距離：

tf.sqrt(tf.reduce_sum(tf.square(tf.expand_dims(A, 1) - B)), 2))