Làm thế nào để so sánh hai tenxơ trong PyTorch?

Để so sánh hai phần tử tensors trong PyTorch, chúng tôi sử dụng torch.eq () phương pháp. Nó so sánh các phần tử tương ứng và trả về "True" nếu hai phần tử giống nhau, nếu không, nó sẽ trả về "False" . Chúng ta có thể so sánh hai bộ căng có kích thước giống nhau hoặc khác nhau, nhưng kích thước của cả hai bộ căng phải phù hợp với kích thước không phải là đơn vị.

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ó.

• Tạo một tensor PyTorch và in nó.

• Tính toán torch.eq (input1, input2) . Nó trả về một tensor "True" và / hoặc "Sai" . Nó so sánh phần tử tensor khôn ngoan và trả về True nếu các phần tử tương ứng bằng nhau, nếu không nó trả về False.

• In tensor trả lại.

Ví dụ 1

Chương trình Python sau đây chỉ ra cách so sánh hai tăng tốc 1-D.

# import necessary library
import torch

# Create two tensors
T1 = torch.Tensor([2.4,5.4,-3.44,-5.43,43.5])
T2 = torch.Tensor([2.4,5.5,-3.44,-5.43, 43])

# print above created tensors
print("T1:", T1)
print("T2:", T2)

# Compare tensors T1 and T2 element-wise
print(torch.eq(T1, T2))

Đầu ra

T1: tensor([ 2.4000, 5.4000, -3.4400, -5.4300, 43.5000])
T2: tensor([ 2.4000, 5.5000, -3.4400, -5.4300, 43.0000])
tensor([ True, False, True, True, False])

Ví dụ 2

Chương trình Python sau đây chỉ ra cách so sánh hai cấp độ căng 2-D.

# import necessary library
import torch

# create two 4x3 2D tensors
T1 = torch.Tensor([[2,3,-32],
                  [43,4,-53],
                  [4,37,-4],
                  [3,75,34]])
T2 = torch.Tensor([[2,3,-32],
                  [4,4,-53],
                  [4,37,4],
                  [3,-75,34]])

# print above created tensors
print("T1:", T1)
print("T2:", T2)

# Conpare tensors T1 and T2 element-wise
print(torch.eq(T1, T2))

Đầu ra

T1: tensor([[ 2., 3., -32.],
            [ 43., 4., -53.],
            [ 4., 37., -4.],
            [ 3., 75., 34.]])
T2: tensor([[ 2., 3., -32.],
            [ 4., 4., -53.],
            [ 4., 37., 4.],
            [ 3., -75., 34.]])
tensor([[ True, True, True],
         [False, True, True],
         [ True, True, False],
         [ True, False, True]])

Ví dụ 3

Chương trình Python sau đây chỉ ra cách so sánh một tensor 1-D với một phần tử 2-Dtensor khôn ngoan.

# import necessary library
import torch

# Create two tensors
T1 = torch.Tensor([2.4,5.4,-3.44,-5.43,43.5])
T2 = torch.Tensor([[2.4,5.5,-3.44,-5.43, 7],
                  [1.0,5.4,3.88,4.0,5.78]])

# Print above created tensors
print("T1:", T1)
print("T2:", T2)

# Compare the tensors T1 and T2 element-wise
print(torch.eq(T1, T2))

Đầu ra

T1: tensor([ 2.4000, 5.4000, -3.4400, -5.4300, 43.5000])
T2: tensor([[ 2.4000, 5.5000, -3.4400, -5.4300, 7.0000],
            [ 1.0000, 5.4000, 3.8800, 4.0000, 5.7800]])
tensor([[ True, False, True, True, False],
         [False, True, False, False, False]])