yuns
torch.diag() 본문
반응형
n x n matrix에서 대각행렬에 해당하는 부분을 반환해줌 1 x n차원으로 반환해줌
반응형
'pytorch module 정리' 카테고리의 다른 글
| pytorch 설치하기 (0) | 2021.06.16 |
|---|---|
| MultiLabelSoftMarginLoss (0) | 2021.05.13 |
| F.log_softmax() (0) | 2021.04.21 |
| torchvision.datasets (cococaptions) (0) | 2021.04.20 |
'pytorch module 정리' Related Articles
more
Comments