반응형 분류 전체보기734 [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를.. 2024. 9. 19. 이전 1 ··· 234 235 236 237 238 239 240 ··· 734 다음 반응형