【发布时间】：2021-03-17 17:34:32
【问题描述】：
当我为张量动作执行 print("action shape:, ", action.shape) 时，我得到了 (64,)。和 (1, 64) 一样吗？以及如何将其大小重塑为 (64,1)？
【问题讨论】：
【发布时间】：2021-03-17 17:34:32
【问题描述】：
从技术上讲，它不是相同的形状，并且在 pytorch 中，如果您有需要 (64,) 形状的东西，但您给了它 (1,64) 但很容易将其更改为 (64 ,) 通过挤压它。要将其重塑为 (64, 1) 的大小，您可以这样做
action = action.unsqueeze(1)
# or
action = action.view(-1, 1)
两者都可以，但我会推荐第一个。
【讨论】：