Monday 21 October 2019

torch resize using view

Resizing: If you want to resize/reshape tensor, you can use torch.view:

torch.Size([4, 4]) torch.Size([16]) torch.Size([2, 8])

No comments:

Post a Comment