텐서에서 요소별 빼기를 수행하려면 torch.sub()를 사용할 수 있습니다. PyTorch의 방법. 텐서의 해당 요소를 뺍니다. 다른 텐서에서 스칼라나 텐서를 뺄 수 있습니다. 차원이 같거나 다른 텐서에서 텐서를 뺄 수 있습니다. 최종 텐서의 차원은 고차원 텐서의 차원과 동일합니다.
단계
-
필요한 라이브러리를 가져옵니다. 다음 모든 Python 예제에서 필수 Python 라이브러리는 torch입니다. . 이미 설치했는지 확인하십시오.
-
둘 이상의 PyTorch 텐서를 정의하고 인쇄하십시오. 스칼라 수량을 빼려면 정의하십시오.
-
torch.sub()를 사용하여 다른 텐서에서 스칼라 또는 텐서를 뺍니다. 값을 새 변수에 할당합니다. 텐서에서 스칼라 수량을 뺄 수도 있습니다. 이 방법을 사용하여 텐서를 빼도 원래 텐서는 변경되지 않습니다.
-
최종 텐서를 인쇄합니다.
예시 1
여기에서는 텐서에서 스칼라 수량을 빼는 Python 3 프로그램이 있습니다. 동일한 작업을 수행하는 세 가지 다른 방법이 표시됩니다.
# Python program to perform element-wise subtraction
# import the required library
import torch
# Create a tensor
t = torch.Tensor([1.5, 2.03, 3.8, 2.9])
print("Original Tensor t:\n", t)
# Subtract a scalar value to a tensor
v = torch.sub(t, 5.60)
print("Element-wise subtraction result:\n", v)
# Same result can also be obtained as below
t1 = torch.Tensor([5.60])
w = torch.sub(t, t1)
print("Element-wise subtraction result:\n", w)
# Other way to do above operation
t2 = torch.Tensor([5.60,5.60,5.60,5.60])
x = torch.sub(t, t2)
print("Element-wise subtraction result:\n", x) 출력
Original Tensor t: tensor([1.5000, 2.0300, 3.8000, 2.9000]) Element-wise subtraction result: tensor([-4.1000, -3.5700, -1.8000, -2.7000]) Element-wise subtraction result: tensor([-4.1000, -3.5700, -1.8000, -2.7000]) Element-wise subtraction result: tensor([-4.1000, -3.5700, -1.8000, -2.7000])
예시 2
다음 프로그램은 2차원 텐서에서 1차원 텐서를 빼는 방법을 보여줍니다.
# Import necessary library
import torch
# Create a 2D tensor
T1 = torch.Tensor([[8,7],[4,5]])
# Create a 1-D tensor
T2 = torch.Tensor([10, 5])
print("T1:\n", T1)
print("T2:\n", T2)
# Subtract 1-D tensor from 2-D tensor
v = torch.sub(T1, T2)
print("Element-wise subtraction result:\n", v) 출력
T1: tensor([[8., 7.], [4., 5.]]) T2: tensor([10., 5.]) Element-wise subtraction result: tensor([[-2., 2.], [-6., 0.]])
예시 3
다음 프로그램은 1D 텐서에서 2D 텐서를 빼는 방법을 보여줍니다.
# Python program to subtract 2D tensor from 1D tensor
# Import the library
import torch
# Create a 2D tensor
T1 = torch.Tensor([[1,2],[4,5]])
# Create a 1-D tensor
T2 = torch.Tensor([10, 5])
print("T1:\n", T1)
print("T2:\n", T2)
# Subtract 2-D tensor from 1-D tensor
v = torch.sub(T2, T1)
print("Element-wise subtraction result:\n", v) 출력
T1: tensor([[1., 2.], [4., 5.]]) T2: tensor([10., 5.]) Element-wise subtraction result: tensor([[9., 3.], [6., 0.]])
최종 텐서는 2D 텐서임을 알 수 있습니다.
예시 4
다음 프로그램은 2D 텐서에서 2D 텐서를 빼는 방법을 보여줍니다.
# import the library
import torch
# Create two 2-D tensors
T1 = torch.Tensor([[8,7],[3,4]])
T2 = torch.Tensor([[0,3],[4,9]])
print("T1:\n", T1)
print("T2:\n", T2)
# Subtract above two 2-D tensors
v = torch.sub(T1,T2)
print("Element-wise subtraction result:\n", v) 출력
T1: tensor([[8., 7.], [3., 4.]]) T2: tensor([[0., 3.], [4., 9.]]) Element-wise subtraction result: tensor([[ 8., 4.], [-1., -5.]])