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

139 votes184 views6 answers
Anaconda create new environment
Python
1 vote182 views1 answer
How to create text file with python and store a dictionary
Python
-1 vote168 views2 answers
Print tables in python
Python
2 votes182 views1 answer
Check the os in python
Python
1 vote169 views1 answer
Import numpy financial python
Python
8 votes193 views3 answers
How to set default python version in macos
Python
7 votes174 views2 answers
Python check for duplicate
Python
1 vote171 views1 answer
How to custom page not found in django
Python
-2 votes171 views1 answer
No connection could be made because the target machine actively refused it python
Python
1 vote227 views1 answer
Open a filename starting with in python
Python