파이토치 정수 랜덤 생성
파이토치 정수 랜덤 생성 # low 에서 high 사이 정수를 size 크기 만큼 생성 torch.randint(low, high, size) # 예시 : torch.randint(0, 10, [2,3]) # out : tensor([[9, 9, 0], # [5, 2, 0]]) [참고] # size 는 참고 용입니다. size = [3,2,4] torch.rand(size) # 0~1 사이 난수 생성(균일분포) torch.randn(size) # 0~1 사이 난수 생성(가우시안분포)
2023. 8. 29.
while와 iter 메서드를 활용하여 파이토치 데이터 로더 5개의 데이터 까지만 확인해 보는 방법
while와 iter 메서드를 활용하여 파이토치 데이터 로더 5개의 데이터 까지만 확인해 보는 방법¶ In [1]: # 패키지 불러오기 import torch from torch.utils.data import DataLoader, Dataset In [2]: # 0에서 100사이 정수로 샘플 데이터 만들기 data_size = 100 sequence_length = 2 vector_size = 3 data = torch.randint(0, 100, [data_size, sequence_length, vector_size]) data[:2] Out[2]: tensor([[[24, 78, 16], [78, 63, 12]], [[98, 42, 13], [27, 62, 49]]]) In [3]: # 샘플 데이..
2023. 8. 28.