/Python/Pytorch tensor change dimension order

Pytorch tensor change dimension order

Pytorch tensor change dimension order

a = torch.rand(1,2,3,4)
print(a.transpose(0,3).transpose(1,2).size())
print(a.permute(3,2,1,0).size())

2

Source: discuss.pytorch.org

Similar codes

0 vote161 views1 answer
Hide password in python
Python
2 votes182 views1 answer
Change pygame window title
Python
3 votes220 views2 answers
Initialise a 2d array python
Python
0 vote176 views2 answers
How to input a string character into a numpy zeros imatrix n python
Python
0 vote174 views1 answer
.comments.all order django
Python
0 vote236 views1 answer
Python json random number generator
Python
0 vote187 views1 answer
SparkSession pyspark
Python
1 vote177 views2 answers
Convert string representation of dict to dict python
Python
0 vote156 views1 answer
Python identify array
Python
0 vote173 views1 answer
Pytorch calculate mse mae
Python

Share