[pytorch, 파이토치] torch.zeros 사용법 특정 사이즈를 갖는 0으로 구성된 텐서 구하기

어떤 pytorch tensor A가 있다고 하자. 이 A와 같은 size를 갖고 0으로 구성된 tensor zeros를 구하고 싶다. 그럴 때 torch.zeros를 사용하면 된다.

목차

torch.zeros 사용법

import torch
A = torch.randn(3,5,2)
print(A.size()) #torch.Size([3, 5, 2])

zeros = torch.zeros(A.size())
print(zeros)
#tensor([[[0., 0.],
#        [0., 0.],
#        [0., 0.],
#        [0., 0.],
#        [0., 0.]],

#       [[0., 0.],
#        [0., 0.],
#        [0., 0.],
#        [0., 0.],
#        [0., 0.]],

#       [[0., 0.],
#        [0., 0.],
#        [0., 0.],
#        [0., 0.],
#        [0., 0.]]])
         
print(zeros.size()) #torch.Size([3, 5, 2])

위와 같이 tensor A를 만든다. 

그런 후에 torch.zeros의 입력변수로 A.size()를 넣는다.

확인해보니 0으로 구성되어있고 A와 size가 같은 tensor가 만들어진다.