在 python 中重塑这个错误是什么？

Question

我有 torch.Size([1, 64, 32, 32]) 的 act['a'] 和 torch.Size([1, 128, 16, 16]) 的 act['b'] 和 act torch.Size([1, 256, 8, 8]) 的 ['c']。 我有：

target = {'a': [], 'b': [], 'c': []}
for i in range(800):
                target['a'].append(act['a'])
                target['b'].append(act['b'])
                target['c'].append(act['c'])

所以现在 target['a'] 的大小是 [800, 1, 64, 32, 32]。如何将其重塑为 [1, 64, 800, 32, 32]？

我试过了

tf.reshape(target['a'], [1, 64, 800,32,32])

它给了我这个错误，我截图了：

Answer 1

这是你所期望的吗？

a = tf.zeros([1, 64, 32, 32])
target = {'a': [a] * 800}
m = tf.convert_to_tensor(target['a'])
n = tf.reshape(m, [1, 64, 800, 32, 32])

>>> m.shape
TensorShape([800, 1, 64, 32, 32])

>>> n.shape
TensorShape([1, 64, 800, 32, 32])