zeros1 [Pytorch] - Ones and Zeros Ones and Zeros 이번에는 Ones and Zeros에 대하여 언급하고자 합니다. 아주 쉬운 개념이지만 ones_like의 경우는 똑같은 Size에 똑같은 Shape에 1로 가득 차게 됩니다. Zeors도 마찬가지 입니다. 사실 이러한 코드는 사용하기에는 매우 쉽지만, 중요한것은 device도 똑같이 가게 된다는 것입니다. GPU에서 Tensor를 선언하게 되는 상황이 생길때는 다른 모델이 GPU에서 돌아가고, CPU에서 텐서를 만들고 CPU, GPU텐서를 더하게 되면 에러가 발생하게 되는데, 이러한 에러를 막기위해 ones_like, zeros_like를 사용하게 되면 같은 Device로 Tensor를 선언해주어 이러한 에러를 막을 수 있겠습니다. 2021. 10. 22. 이전 1 다음
