torch.eye(n,...) : 대각선에 1이 있고 다른 곳에는 0으로 채워진, 크기가 nxn인 텐서 만들어줘

dusruddl2·2022년 9월 22일

0

pytorch tutorial

목록 보기

3/4

torch.eye(n,...)
: 대각선에 1이 있고 다른 곳에는 0으로 채워진, 크기가 nxn인 텐서 만들어줘

i = torch.eye(3)
print(i)

정리된 글은 https://dusruddl2.tistory.com/로 이동

이전 포스트

torch.ones, zeros, full, empty에 대하여

다음 포스트

torch.arange(n): 0~(n-1)까지 갖는 텐서를 만들어줘

0개의 댓글