张量流张量的元素运算

Question

我是张力流的新手，需要做一些在任何其他情况下都不会那么难的事情。

我需要采用形状（？，224,224,3）的张量并对每个元素应用一个操作，保持大小。我已经尝试过tf.map_fn（myFunc，data），但似乎传入myFunc的是原始张量的一个子集。

有没有简单的方法可以说“对张量中的所有事情做这个操作”？

Answer 1

也许你需要的是broad casting semantics。

Answer 2

最简单的解决方案可能是对输入进行矢量化，应用函数，然后进行非向量化。像这样的东西

y_shape = tf.shape( y )
y_vec = tf.reshape( y, [-1] )
fy_vec = tf.map_fn( fn, y_vec )
fy = tf.reshape( fy_vec, y_shape )

只要你的函数将标量映射到标量，这就可以工作。