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)

兩者都可以，但我會推薦第一個。