[Pytorch] 임의의 배열/텐서 만들기

본 글에서 사용한 pytorch의 버전은 1.13.0이다.

종종 pytorch을 사용하다 보면 임의의 배열을 만들 필요가 생긴다. 그럴때 유용한 코드를 공유한다.

 

zeros

0으로 채워진 주어진 사이즈의 tensor를 만든다.

a = torch.zeros((2,2))
print(a.shape)
#torch.Size([2, 2])
print(a)
#tensor([[0., 0.],
#        [0., 0.]])

 

ones

1로 채워진 주어진 사이즈의 tensor를 만든다.

a = torch.ones((2, 2))
print(a.shape)
print(a)
#torch.Size([2, 2])
#tensor([[1., 1.],
#        [1., 1.]])

empty

초기화 되지 않은 값으로 주어진 사이즈의 tensor를 만든다.

a = torch.empty((2, 2))
print(a.shape)
#torch.Size([2, 2])

 

References

https://pytorch.org/docs/stable/torch.html

torch — PyTorch 2.0 documentation