Pytorch view -1 meaning

View(-1) in pytorch

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

Pytorch view -1 meaning

If there is any situation that you don't know how many rows you want but are sure of the number of columns, then you can specify this with a -1. (Note that you can extend this to tensors with more dimensions. Only one of the axis value can be -1). This is a way of telling the library: "give me a tensor that has these many columns and you compute the appropriate number of rows that is necessary to make this happen".

Similar codes

24 votes50 views6 answers
Python sort dictionary by key
Python
14 votes42 views1 answer
Dataframe row
Python
1 vote52 views1 answer
Python index 2d array
Python
0 vote43 views1 answer
How to print keys and values of dictionary together in python?
Python
4 votes68 views4 answers
Convert image to grayscale opencv
Python
13 votes35 views3 answers
To check pip version
Python
0 vote48 views1 answer
Discord py message link
Python
0 vote63 views1 answer
Which of the following statements is used to create an empty set in python?
Python
0 vote1 view1 answer
Pandas code
Python
0 vote44 views1 answer
Ignore pytest errors or warnings
Python