Tensorflow：如何以“笛卡尔积”方式减去2个张量？

Question

我想做这样的事情：

v = tf.get_variable('v', [N])
p = tf.placeholder(shape=[None])
tmp = []
for element_p in p:
   tmp.append(element_p - v)
ret = tf.stack(tmp, axis=0)
# ret.get_shape().as_list() = [None, N]

这在Tensorflow中是否可行？

Answer 1

利用广播：

import tensorflow as tf

N = ...
v = tf.get_variable('v', [N])
p = tf.placeholder(shape=[None])
ret = p[:, tf.newaxis] - v[tf.newaxis, :]