pytorch random, transpose, permute

ssm·2024년 1월 24일

pytorch

목록 보기
1/3

random

rand 0부터 1 사이 (균등분포)
randn 평균 0, 표준편차 1(정규분포)
randint 정수를 뽑음 (균등분포)

np.random.seed(42)	# seed 고정하면 동일하게 생성
np.random.rand(5)	# 0부터 1사이의 난수 5개
np.random.randn(5) 	# 표준편차가 1인 난수 5개 from 정규분포(0, 1)
np.random.randint(0, 10, size=5)	# 0부터 10미만의 정수 난수 5개

transpose vs permute

transpose dim 두 개 서로 바꾸기
permute dim 여러 개 순서 바꾸기

x = torch.rand(5, 3, 8)
y = x.transpose(0, 2)  # [8, 3, 5]
z = x.permute(1, 2, 0)  # [3, 8, 5]

기타

.item()
텐서 값 만을 가져옴

profile
내 뇌의 외장하드

0개의 댓글