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 vote251 views2 answers
How to run turtle in python
Python
1 vote301 views1 answer
Django model query add annotation field to show duplicate count
Python Django
0 vote234 views1 answer
Python how to switch between true and false
Python
28 votes263 views3 answers
Datetime one month ago python
Python
19 votes272 views1 answer
Python string cut right
Python
0 vote246 views1 answer
No default language could be detected for django app
Python
0 vote296 views1 answer
Pandas como eliminar filas con valores no nulos en una columna
Python
1 vote241 views1 answer
Filter dataframe columns vy a list of columns
Python
0 vote259 views1 answer
Python modules screen
Python
-1 vote253 views2 answers
Pandas dataframe apply function with multiple arguments
Python