반응형 분류 전체보기809 [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 ··· 309 310 311 312 313 314 315 ··· 809 다음 반응형