Chúng tôi có thể sử dụng torch.add () để thực hiện phép cộng phần tử khôn ngoan trên tenxơ trong PyTorch. Nó thêm các phần tử tương ứng của tenxơ. Chúng ta có thể thêm một đại lượng vô hướng hoặc tensor vào một tensor khác. Chúng tôi có thể thêm các tenxơ có kích thước giống nhau hoặc khác nhau. Kích thước của tensor cuối cùng sẽ giống với kích thước của tensor thứ nguyên cao hơn.
Các bước
-
Nhập thư viện được yêu cầu. Trong tất cả các ví dụ Python sau, thư viện Python bắt buộc là torch . Đảm bảo rằng bạn đã cài đặt nó.
-
Xác định hai hoặc nhiều bộ căng PyTorch và in chúng. Nếu bạn muốn lấy một đại lượng vô hướng, hãy xác định nó.
-
Thêm hai hoặc nhiều bộ căng bằng cách sử dụng torch.add () và gán giá trị cho một biến mới. Bạn cũng có thể thêm một đại lượng vô hướng vào tensor. Việc thêm các dây căng bằng cách sử dụng phương pháp này không tạo ra bất kỳ thay đổi nào trong các dây căng ban đầu.
-
In căng cuối cùng.
Ví dụ 1
Chương trình Python sau đây cho thấy cách thêm một số lượng vô hướng vào atensor. Chúng tôi thấy có ba cách khác nhau để thực hiện cùng một nhiệm vụ.
# Python program to perform element-wise Addition
# import the required library
import torch
# Create a tensor
t = torch.Tensor([1,2,3,2])
print("Original Tensor t:\n", t)
# Add a scalar value to a tensor
v = torch.add(t, 10)
print("Element-wise addition result:\n", v)
# Same operation can also be done as below
t1 = torch.Tensor([10])
w = torch.add(t, t1)
print("Element-wise addition result:\n", w)
# Other way to perform the above operation
t2 = torch.Tensor([10,10,10,10])
x = torch.add(t, t2)
print("Element-wise addition result:\n", x) Đầu ra
Original Tensor t: tensor([1., 2., 3., 2.]) Element-wise addition result: tensor([11., 12., 13., 12.]) Element-wise addition result: tensor([11., 12., 13., 12.]) Element-wise addition result: tensor([11., 12., 13., 12.])
Ví dụ 2
Chương trình Python sau đây chỉ ra cách thêm các tensors 1D và 2D.
# Import the library
import torch
# Create a 2-D tensor
T1 = torch.Tensor([[1,2],[4,5]])
# Create a 1-D tensor
T2 = torch.Tensor([10]) # also t2 = torch.Tensor([10,10])
print("T1:\n", T1)
print("T2:\n", T2)
# Add 1-D tensor to 2-D tensor
v = torch.add(T1, T2)
print("Element-wise addition result:\n", v) Đầu ra
T1: tensor([[1., 2.], [4., 5.]]) T2: tensor([10.]) Element-wise addition result: tensor([[11., 12.], [14., 15.]])
Ví dụ 3
Chương trình sau đây trình bày cách thêm bộ căng 2D.
# Import the library
import torch
# create two 2-D tensors
T1 = torch.Tensor([[1,2],[3,4]])
T2 = torch.Tensor([[0,3],[4,1]])
print("T1:\n", T1)
print("T2:\n", T2)
# Add the above two 2-D tensors
v = torch.add(T1,T2)
print("Element-wise addition result:\n", v) Đầu ra
T1: tensor([[1., 2.], [3., 4.]]) T2: tensor([[0., 3.], [4., 1.]]) Element-wise addition result: tensor([[1., 5.], [7., 5.]])