tensor를 float()로 만든다.
- torch.mean(input)
- input.mean()
두 가지 방법이 있다.
- torch.tensor는 객체 자체를 계산시키는 방법이다.
- torch.mean은 함수로 인자값을 넣어 계산한다.
https://frieden1946.tistory.com/556?category=952749 에서 나온
torch.nn 과 torch.nn.functional
차이점과 동일하다고 할 수 있다.
'AI > PyTorch' 카테고리의 다른 글
[PyTorch] CNN 설계 10.저장된 모델 불러오기 및 test (0) | 2021.06.29 |
---|---|
[PyTorch] torch.save(model.state_dict(), PATH) / torch.load(PATH) (0) | 2021.06.29 |
[PyTorch] torch.tensor.float(), torch.mean() (0) | 2021.06.29 |
[PyTorch] torch.max(outputs, 1) (0) | 2021.06.29 |
[PyTorch] Conv1d, Conv2d 차이점 (0) | 2021.06.29 |