PyTorch에서 두 개의 텐서에 대해 요소별 분할을 수행하려면 torch.div()를 사용할 수 있습니다. 방법. 첫 번째 입력 텐서의 각 요소를 두 번째 텐서의 해당 요소로 나눕니다. 텐서를 스칼라로 나눌 수도 있습니다. 텐서는 차원이 같거나 다른 텐서로 나눌 수 있습니다. 최종 텐서의 차원은 고차원 텐서의 차원과 동일합니다. 1D 텐서를 2D 텐서로 나누면 최종 텐서는 2D 텐서가 됩니다.
단계
-
필요한 라이브러리를 가져옵니다. 다음 모든 Python 예제에서 필수 Python 라이브러리는 torch입니다. . 이미 설치했는지 확인하십시오.
-
둘 이상의 PyTorch 텐서를 정의하고 인쇄하십시오. 텐서를 스칼라로 나누고 싶다면 스칼라를 정의하세요.
-
torch.div()를 사용하여 텐서를 다른 텐서 또는 스칼라로 나눕니다. 값을 새 변수에 할당합니다. 이 방법을 사용하여 텐서를 나누어도 원래 텐서는 변경되지 않습니다.
-
최종 텐서를 인쇄합니다.
예시 1
# Python program to perform element-wise division
# import the required library
import torch
# Create a tensor
t = torch.Tensor([2, 3, 5, 9])
print("Original Tensor t:\n", t)
# Divide a tensor by a scalar 4
v = torch.div(t, 4)
print("Element-wise division result:\n", v)
# Same result can also be obtained as below
t1 = torch.Tensor([4])
w = torch.div(t, t1)
print("Element-wise division result:\n", w)
# other way to do above operation
t2 = torch.Tensor([4,4,4,4])
x = torch.div(t, t2)
print("Element-wise division result:\n", x) 출력
Original Tensor t: tensor([2., 3., 5., 9.]) Element-wise division result: tensor([0.5000, 0.7500, 1.2500, 2.2500]) Element-wise division result: tensor([0.5000, 0.7500, 1.2500, 2.2500]) Element-wise division result: tensor([0.5000, 0.7500, 1.2500, 2.2500])
예시 2
다음 Python 프로그램은 2D 텐서를 1D 텐서로 나누는 방법을 보여줍니다.
# import the required library
import torch
# Create a 2D tensor
T1 = torch.Tensor([[3,2],[7,5]])
# Create a 1-D tensor
T2 = torch.Tensor([10, 8])
print("T1:\n", T1)
print("T2:\n", T2)
# Divide 2-D tensor by 1-D tensor
v = torch.div(T1, T2)
print("Element-wise division result:\n", v) 출력
T1: tensor([[3., 2.], [7., 5.]]) T2: tensor([10., 8.]) Element-wise division result: tensor([[0.3000, 0.2500], [0.7000, 0.6250]])
예시 3
다음 Python 프로그램은 1D 텐서를 2D 텐서로 나누는 방법을 보여줍니다.
# Python program to dive a 1D tensor by a 2D tensor
# import the required 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)
# Divide 1-D tensor by 2-D tensor
v = torch.div(T2, T1)
print("Division 1D tensor by 2D tensor result:\n", v) 출력
T1: tensor([[8., 7.], [4., 5.]]) T2: tensor([10., 5.]) Division 1D tensor by 2D tensor result: tensor([[1.2500, 0.7143], [2.5000, 1.0000]])
최종 텐서는 2D 텐서임을 알 수 있습니다.
예시 4
다음 Python 프로그램은 2D 텐서를 2D 텐서로 나누는 방법을 보여줍니다.
# import necessary library
import torch
# Create two 2-D tensors
T1 = torch.Tensor([[8,7],[3,4]])
T2 = torch.Tensor([[0,3],[4,9]])
# Print the above tensors
print("T1:\n", T1)
print("T2:\n", T2)
# Divide T1 by T2
v = torch.div(T1,T2)
print("Element-wise division result:\n", v) 출력
T1: tensor([[8., 7.], [3., 4.]]) T2: tensor([[0., 3.], [4., 9.]]) Element-wise division result: tensor([[ inf, 2.3333], [0.7500, 0.4444]])