[pytorch, 파이토치] torch.zeros_like 0으로 구성된 텐서 만들기

이번 글에서는 pytorch에서 어떤 tensor A가 있을때 A와 사이즈가 똑같고 0으로 구성된 텐서를 만드는 방법에 대해 알아보자. 그러기 위해서 torch.zeros_like를 사용하면 된다.

목차

torch.zeros_like 사용법

torch.zeros_like(tensor)

위와 같이 torch.zeros_like에 입력으로 tensor를 넣어주면 된다.

 

torch.zeros_like 사용 예시

import torch
tensor = torch.randn(3,2,3)
print(tensor)

#tensor([[[-0.1230, -1.4075,  1.7624],
#         [-1.6692,  1.2652, -1.2060]],
#
#        [[-0.5205, -2.1617,  0.0347],
#         [ 1.0592, -0.1771, -1.0436]],
#
#        [[ 0.8409,  1.6975, -0.7908],
#         [-0.1894, -0.5630,  0.7894]]])

 

위와 같이 torch 모듈을 불러오고 3,2,3 크기의 tensor를 만들었다.

zeros = torch.zeros_like(tensor)
print(zeros)

#tensor([[[0., 0., 0.],
#        [0., 0., 0.]],

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

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

위와 같이 torch.zeros_like에 입력으로 tensor를 넣어주었다.

print를 해보니 0으로 구성된 tensor가 만들어졌다.

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

위와 같이 tensor의 사이즈가 동일하다는 것을 확인할 수 있다.