[pytorch] torch.ones 1값으로 구성된 텐서 생성하기

모든값이 1로 구성된 텐서를 생성하는 방법중에 하나인 torch.ones에 대해 알아보자.

torch.ones 사용법

torch.ones(size)

위와 같이 입력하면 원하는 사이즈를 갖는 1로 구성된 텐서가 만들어진다.

torch.ones 사용 예시

입력으로 정수를 넣는 경우에는 정수길이만큼의 텐서가 만들어진다.

test = torch.ones(3)
print(test)
#tensor([1., 1., 1.])

tuple로 사이즈를 넣어도 된다.

test = torch.ones(1,3,2)
print(test)
#tensor([[[1., 1.],
#         [1., 1.],
#         [1., 1.]]])

어떤 tensor가 있을 때 이것과 같은 크기의 tensor를 만들고 싶다면 이 tensor의 size를 입력하면 된다.

tensor = torch.randn(1,2,3,4)
test = torch.ones(tensor.size())
print(test)
#tensor([[[[1., 1., 1., 1.],
#          [1., 1., 1., 1.],
#          [1., 1., 1., 1.]],
#
#         [[1., 1., 1., 1.],
#         [1., 1., 1., 1.],
#          [1., 1., 1., 1.]]]])

torch.nn.DataParallel을 일반 모델로 전환하기

안녕하세요. 파이토치 (pytorch)를 사용하다보면 GPU를 여러개 사용하기 위하여 torch.nn.DataParallel을 이용해 모델을 불러오곤 합니다. 물론 DataParallel을 이용하면 여러 GPU를 사용할 수 있기 때문에 편

qwertyuioop.tistory.com

[pytorch, 파이토치] torch.ones_like, 같은 사이즈의 1로 구성된 텐서 만들기

print(ones.size()) #torch.Size([3, 2, 4])print(tensor.size()) #torch.Size([3, 2, 4])어떤 텐서와 같은 size를 갖으면서 구성요소가 모두 1인 텐서를 만들때는 torch.ones_like를 사용하면 됩니다. 이번 글에서는 torch.ones_li

qwertyuioop.tistory.com

[pytorch] torch.ones 1값으로 구성된 텐서 생성하기

모든값이 1로 구성된 텐서를 생성하는 방법중에 하나인 torch.ones에 대해 알아보자.목차torch.ones 사용법torch.ones(size)위와 같이 입력하면 원하는 사이즈를 갖는 1로 구성된 텐서가 만들어진다.torch.on

qwertyuioop.tistory.com

[python] itertools.product 으로 조합 만들기

python 에서 리스트를 구성하는 원소간의 조합을 만들고 싶을 때 여러 방법이 있지만 itertools.product 를 사용하는 방법이 있다. itertools.product 설명 및 예시 itertools을 이용하면 두개 이상의 리스트에

qwertyuioop.tistory.com

[python] 파이썬으로 난수, 랜덤넘버 생성하기 random.random() 함수 사용법

파이썬 (python)으로 난수, 랜덤넘버 생성하는 방법 중의 하나인 random.random() 함수 사용법에 대해 알아보겠습니다. 이번 글에서는 random.random() 함수 사용법 그리고 이것을 이용해서 일정 범위내에

qwertyuioop.tistory.com

[pytorch, 파이토치] torch.ones_like, 같은 사이즈의 1로 구성된 텐서 만들기

qwertyuioop.tistory.com