[pytorch, 파이토치] torch.zeros 사용법 특정 사이즈를 갖는 0으로 구성된 텐서 구하기
어떤 pytorch tensor A가 있다고 하자. 이 A와 같은 size를 갖고 0으로 구성된 tensor zeros를 구하고 싶다. 그럴 때 torch.zeros를 사용하면 된다.목차torch.zeros 사용법import torchA = 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.],# ..
2024. 9. 21.