- 수정 중
텐서 연산
import math
import torch
a = torch.tensor([0.3,-0.6,0.8])
print(a * 2 -1) # tensor([-0.4000, -2.2000, 0.6000])
print(torch.abs(a)) # tensor([0.3000, 0.6000, 0.8000])
print(torch.ceil(a)) # tensor([1., -0., 1.])
print(torch.floor(a)) # tensor([ 0., -1., 0.])
print(torch.clamp(a,-0.5,0.5)) # tensor([ 0.3000, -0.5000, 0.5000])
- abs 절대값
- ceil 올림
- floor 내림
- clamp(적용할 텐서, x ,y) -> x,y사이의 요소들 출력 그 범위를 넘어서면 x,y로 출력한다.
Tensor 연산
a = torch.tensor([1,2,3])
b = torch.tensor([4,5,6])
print("a")
print("b")
- 행렬곱과 마찬가지로 세개다 같은 연산 다른 표현일 뿐이다.
요소를 하나로 만들고 그 요소를 파이썬 자료형으로 바꾸는 방법
agg = tensor.sum()
agg_item = agg.item() # tensor에서 pythone 의 float 형태로 바꿔줌
print(agg_item, type(agg_item))
'Pytorch' 카테고리의 다른 글
torchvision.transforms 함수 (0) | 2022.11.19 |
---|---|
PyTorch - 1) 텐서(Tensor) (0) | 2021.12.29 |
PyTorch 시작하기 (0) | 2021.12.28 |