공부방

1. Tensor 만들기 가장 먼저 항상 사용되는 라이브러리인 torch와 numpy를 임포트 import numpy as np import torch 임의의 값으로 초기화된 텐서를 만드는 rand(), 차원과 데이터 타입을 지정해 줄 수 있다. x = torch.rand(2, 3, dtype=torch.float16) # 2 * 3차원이고 데이터 타입은 float16인 텐서 생성 tensor([[0.8472, 0.3481, 0.9297], [0.2480, 0.0684, 0.0674]], dtype=torch.float16) 1로 초기화된 텐서를 만드는 zeros() (비슷하게 0으로 초기화 시켜주는 zeros도 있다.) x = torch.zeros(4, 2) tensor( [[0., 0.], [0., ..