python-3.x - tensorflow map_fn 单值 Tensor

Question

是否可以在具有单个值的张量上运行 map_fn？

以下工作:

import tensorflow as tf
a = tf.constant(1.0, shape=[3])
tf.map_fn(lambda x: x+1, a)
#output: [2.0, 2.0, 2.0]

然而，这不会:

import tensorflow as tf
b = tf.constant(1.0)
tf.map_fn(lambda x: x+1, b)
#expected output: 2.0

有可能吗？

我究竟做错了什么？

任何提示将不胜感激!

Answer 1

不，这是不可能的。正如您可能看到的那样，它会引发错误:
ValueError: elems must be a 1+ dimensional Tensor, not a scalarmap_fn点是将函数应用于张量的每个元素，因此将其用于标量(单元素)张量是没有意义的。

至于“你做错了什么”:如果不知道你想要实现什么，这很难说。