torch.topk

예측 값에서 argmax가 아닌 top-k에 대한 결과 값을 받고 싶을 때 torch.topk를 사용해 받을 수 있다. 더 자세한 사용 예는 References를 참고.https://www.programcreek.com/python/example/101209

2020년 4월 7일
·
0개의 댓글