Torch view

Torch.view

import torch

x = torch.arange(6)

print(x.view(3, -1))      # inferred size will be 2 as 6 / 3 = 2
# tensor([[ 0.,  1.],
#         [ 2.,  3.],
#         [ 4.,  5.]])

print(x.view(-1, 6))      # inferred size will be 1 as 6 / 6 = 1
# tensor([[ 0.,  1.,  2.,  3.,  4.,  5.]])

print(x.view(1, -1, 2))   # inferred size will be 3 as 6 / (1 * 2) = 3
# tensor([[[ 0.,  1.],
#          [ 2.,  3.],
#          [ 4.,  5.]]])

# print(x.view(-1, 5))    # throw error as there's no int N so that 5 * N = 6
# RuntimeError: invalid argument 2: size '[-1 x 5]' is invalid for input with 6 elements

print(x.view(-1, -1, 3))  # throw error as only one dimension can be inferred
# RuntimeError: invalid argument 1: only one dimension can be inferred

Source: stackoverflow.com

Similar codes

0 vote342 views1 answer
Python django if key exists
Python Django
1 vote361 views1 answer
How to write words on any other apps in python
Python
0 vote328 views1 answer
Align a text python
Python
0 vote328 views1 answer
Valueerror: must have equal len keys and value when setting with an iterable
Python
0 vote330 views1 answer
Oauthlib python error
Python
3 votes346 views1 answer
Ipython autoreload
Python
0 vote336 views1 answer
Disable json dumps encode
Python
16 votes341 views4 answers
How to add path to python in windows
Python
38 votes378 views6 answers
How to create virtual environment in anaconda
Python
0 vote382 views1 answer
Distribute a local Python collection to form an RDD
Python