Boostcamp AI tech 3기/개념 이해

torch.range is deprecated warning -> torch.arange

mcdn 2022. 1. 25. 15:24
반응형

 

 

/usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:7: UserWarning: torch.range is deprecated and will be removed in a future release because its behavior is inconsistent with Python's range builtin. Instead, use torch.arange, which produces values in [start, end).

 

pytorch의 range 함수는 파이썬의 range함수와 동작이 달라 제거될 예정입니다. 따라서 torch.arange함수를 사용해주세요. 

 

실제로 range함수는 torch.range(0,10)을 하면 

tensor([ 0., 1., 2., 3., 4., 5., 6., 7., 8., 9., 10.])

처럼 0~10!!까지 값을 출력한다. 파이썬의 range함수라면 0~9로만 이루어진 리스트를 반환.

 

python range와 다르게 동작하기 때문에 미래에 제거될 예정이라고 함 

 

반면 torch.arange(0,10)을 하면 

tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

range함수처럼 0~9을  돌려주는 것을 알 수 있다. 

 

결론 : torch.arange 함수를 사용하자.  

 

 

반응형