반응형
Pytorch Broadcasting
-
[PyTorch Study]Tensor 자료형 연산2) Tech 2020. 9. 23. 08:18
PyTorch Study 두번째 시간입니다!! 오늘은 Pytorch Tensor 자료형 연산에 대해 알아보겠습니다. tensor 자료형 연산은 numpy ndarray 연산과 유사한점이 굉장히 많습니다. 1. tensor 자료형 연산 # 덧셈 뺄셈, 제곱 import torch t1 = torch.tensor([1,2,3]) t2 = torch.tensor([5,6,7]) t3 = t1+10 t4 = t1 - t2 print(t3, '\n') print(t2**2) print('\n', t4) 보시면 tensor 자료형으로 덧셈, 뺄셈, 제곱 연산이 가능합니다. # Broad casting t1 = torch.tensor([1,2,3]) t5 = torch.tensor([[10, 20, 30], [40,..