Tensor reshaping

1

I need to reshape a tensor with size [3, 32] now I need to change the dimension so that the shape becomes [1, 3, 32] ?

Is there any similar method like reshape in numpy for torch tensors?

Thanks.

1 Like
2

You can do:

x = x.view(1, 3, 32)
2 Likes
3

Thank you so much for the quick reply.

4

You can do: x = x.unsqueeze(0)

3 Likes