pytorch module 정리
torch.diag()
yuuuun
2021. 4. 21. 12:16
반응형
n x n matrix에서 대각행렬에 해당하는 부분을 반환해줌 1 x n차원으로 반환해줌
반응형