在一个无量纲上迭代张量

Question

如果我有一个与批量大小相对应的非第一维的张量，例如：

tensor = tf.placeholder(tf.float32, shape=[None, 256, 256,3],name="placeholder_input")

现在我有了一个张量函数“myfunc”，作用于一个大小为[256256,3]的张量，我想将其应用到批量数量的倍上，从而得到大小为[None，256，256,3]的输出。如果形状不是动态的，我只想：

^{pr2}$

我如何处理动态形状？在

Answer 1

如果您真的想这样做，可以使用tf.map_fn

否则，您可以尝试直接处理原始张量（第一维=无）并在适当的轴上进行操作（无需循环）