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

Question

When I do print("action shape:, ", action.shape) for my tensor action I got (64,). It is the same as (1, 64)? And how do I reshape its size to (64,1)?

Answer 1

Technically it is not the same shape and in pytorch you will get an error if you have things that need a shape of (64,) but you give it (1,64) but it is easy to change it to (64,) by squeezing it. To reshape it to a size of (64, 1) you can do this

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

either will work but I would recommend the first one.