다음은 텐서를 ndarray로 변환한 결과입니다.
[[1 2] [3 4]] [[1 2] [3 4]]
텐서의 인덱스 조작
텐서의 인덱스를 조작하는 방법은 여러 가지가 있습니다. 텐서는 넘파이의 ndarray를 조작하는 것과 유사하게 동작하기 때문에 배열처럼 인덱스를 바로 지정하거나 슬라이스 등을 사용할 수 있습니다. 또한, 텐서의 자료형은 다음과 같습니다.
• torch.FloatTensor: 32비트의 부동 소수점
• torch.DoubleTensor: 64비트의 부동 소수점
• torch.LongTensor: 64비트의 부호가 있는 정수
이외에도 다양한 유형의 텐서가 있습니다.
텐서의 인덱스 조작은 다음과 같은 코드를 이용합니다.
temp = torch.FloatTensor([1, 2, 3, 4, 5, 6, 7]) ------ 파이토치로 1차원 벡터 생성 print(temp[0], temp[1], temp[-1]) ------ 인덱스로 접근 print('------------------------') print(temp[2:5], temp[4:-1]) ------ 슬라이스로 접근
코드에서 -1번 인덱스는 맨 뒤에서부터 시작하는 인덱스를 의미합니다.