我需要能够仅在其最后一个轴上 reshape 张量:
(None, 4) --> (None, 2, 2),
在执行时可以有这样的实例:
(128, 10, 4) --> (128, 10, 2, 2)
(128, 4) --> (128, 2, 2)
是否有直接的解决方案，或者我需要迭代第一个轴(通过排除最后一个轴)，并考虑它可以为“无”的情况？
最佳答案
你可以这样做:
my_tensor = ...
new_shape = tf.concat([tf.shape(my_tensor)[:-1], [2, 2]], axis=0)
my_tensor_reshaped = tf.reshape(my_tensor, new_shape)
关于python - reshape 具有多个未知维度的张量，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/60230863/