torch.zeros를 사용하면 torch.Tensor을 scratch부터 만들어내는 것이고
torch.new_zeros를 사용하면 이전에 사용하던 tensor A와 같은 device를 갖게 된다.
cpu를 사용할 때 일일이 cpu 설정을 해주지 않고 (gpu에 대해서도 마찬가지)
이전 tensor A에서 사용하던 device를 계속 이어받아 생성하게 된다.
|
1
2
3
|
shape = (3,4,5)
A = torch.ones((2,3), dtype=torch.int8).
B = A.new_zeros((shape))
|
cs |
new_zeros 뿐 아니라 new 뒤에 올 수 있는 것은 new_tensor, new_full, new_empty, new_ones, new_zeros가 있다.
'딥러닝 꿀팁' 카테고리의 다른 글
| 딥러닝 iteration 돌다가 중간에 죽었음 뜰 때는 pin_memory를 먼저 보라 (0) | 2024.02.27 |
|---|---|
| Pytorch에서 쓸 때마다 헷갈리는 unfold, fold 정리 (2) | 2024.01.09 |
| for문과 같은 loop문 없이 tensor을 특정 조건으로 훑고 선택 및 변경하기 (1) | 2023.07.10 |
| [파이썬]리스트에서 특정 column 내용 추출하기 (0) | 2023.05.24 |
| 모델을 Evaluation시 필수적으로 넣어줘야 할 것 (0) | 2023.05.16 |
