关于张量形状的pytorch问题。以及如何重塑张量

Question

当我为我的张量动作做 print("action shape:, ", action.shape) 时，我得到了 (64,)。和 (1, 64) 一样吗？以及如何将其大小重塑为 (64,1)？

Answer 1

从技术上讲，它不是相同的形状，在 pytorch 中，如果您有需要形状为 (64,) 的东西，但您给它 (1,64)，但很容易将其更改为 (64) ,) 通过挤压它。要将其重塑为 (64, 1) 的大小，您可以这样做

action = action.unsqueeze(1)
# or
action = action.view(-1, 1)

两者都可以，但我会推荐第一个。