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

dusruddl2·2022년 9월 22일

0

pytorch tutorial

목록 보기

4/4

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

j = torch.arange(10)
print(j)

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

이전 포스트

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

0개의 댓글