유용한 PyTorch 정리

torch.tensor에 대해서 앞에 물결을 붙이는 경우가 있다. 베르누이 분포를 생성행렬 a에 대해 앞에 n개, 뒤에 m개의 값x인 패딩을 반환

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

torch.gather란

간혹 깃헙을 보다보면 torch gether를 볼수가 있는데 어떻게 동작하는 건지 잘 이해되지 않아 정리해본다.input 텐서가 입력으로 주어지고, 차원 dim을 따라서 각 행으로부터 값을 취해, 새로운 텐서를 반환한다. torch.LongTensor를 index로

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

torch.topk

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

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

pytorch 모델 저장 및 불러오기

torch.save: 직렬화된 객체를 디스크에 저장. 직렬화는 pickle을 사용. 모든 종류의 객체의 모델 및 텐서, 딕셔너리 저장 가능torch.load: pickle을 사용하여 저장된 객체 파일들을 역직렬화를 통해 메모리에 올린다. 데이터를 장치에 불러올 때도 사

2020년 3월 24일
·
0개의 댓글