Answers for "squeeze and unsqueeze pytorch"

Python

1

what is squeeze function in pytorch?

import torch
t = torch.tensor([[1,17,7,
                     3,9,10]])
print(t)
>>>tensor([[1,17,7,3,9,10]])
print(torch.squeeze(t))
>>>tensor([ 1, 17,  7,  3,  9, 10])

t = torch.tensor([[[1,17,7,
                     3,9,10]]])
print(t)
>>>tensor([[[1,17,7,3,9,10]]])
print(torch.squeeze(t))
>>>tensor([ 1, 17,  7,  3,  9, 10])

Posted by: Guest on November-26-2020

Code answers related to "squeeze and unsqueeze pytorch"

Code answers related to "Python"

Python Answers by Framework

Django
Flask

Browse Popular Code Answers by Language

Python

Javascript

Whatever

Shell/Bash

CSS

Html

PHP

SQL

Java

C#