我想更改张量的形状/尺寸。 例如,
x = tf.constant([1, 2, 3])
那我想得到y = tf.constant([1,1], [2,2], [3,3]).
如何获得新的y
转换?
我已经运行了代码
y = tf.map_fn(lambda e : (e, e), x)
但是发生了错误:
ValueError: The two structures don't have the same nested structure.
答案 0 :(得分:2)
确保您明确声明了以下类型:
y = tf.map_fn(lambda e : (e, e), x, dtype=(tf.float32, tf.float32))