Computer/코딩 개꿀팁
Pytorch tensor를 더 많이/정확하게 표기하고 싶을 때
벼랑끝과학자
2023. 2. 23. 16:57
torch.set_printoptions()함수를 이용한다.
1. precision : int , 소수점 몇자리까지 나타낼 것인지?
2. threshold : int, : 텐서 엘레멘트가 threshold개 이상일 경우에만 ... 등으로 축약해서 표시하기
대충 default는 1000인데 10000정도로 올리면 앵간하면 다 확인할 수 있다.