Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

จะเปรียบเทียบเทนเซอร์สองตัวใน PyTorch ได้อย่างไร


ในการเปรียบเทียบเมตริกซ์สององค์ประกอบใน PyTorch เราใช้ torch.eq() กระบวนการ. เปรียบเทียบองค์ประกอบที่เกี่ยวข้องและส่งคืน "จริง" หากองค์ประกอบทั้งสองเหมือนกัน มิฉะนั้นจะส่งกลับ "False" . เราสามารถเปรียบเทียบเมตริกซ์สองตัวที่มีขนาดเท่ากันหรือต่างกันได้ แต่ขนาดของเมตริกซ์ทั้งสองจะต้องตรงกันในมิติที่ไม่ใช่แบบเดี่ยว

ขั้นตอน

  • นำเข้าไลบรารีที่จำเป็น ในตัวอย่าง Python ทั้งหมดต่อไปนี้ ไลบรารี Python ที่จำเป็นคือ ไฟฉาย . ตรวจสอบให้แน่ใจว่าคุณได้ติดตั้งแล้ว

  • สร้างเทนเซอร์ PyTorch แล้วพิมพ์ออกมา

  • คำนวณ torch.eq(input1, input2) . ส่งกลับเทนเซอร์ของ "จริง" และ/หรือ "เท็จ" . โดยจะเปรียบเทียบองค์ประกอบเทนเซอร์และคืนค่า True หากองค์ประกอบที่เกี่ยวข้องกัน มิฉะนั้นจะคืนค่าเป็น False

  • พิมพ์เทนเซอร์ที่ส่งคืน

ตัวอย่างที่ 1

โปรแกรม Python ต่อไปนี้แสดงวิธีเปรียบเทียบ 1-D tensorselement-wise

# 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))

ผลลัพธ์

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])

ตัวอย่างที่ 2

โปรแกรม Python ต่อไปนี้แสดงวิธีเปรียบเทียบการตรวจวัดค่าเทนเซอร์แบบ 2 มิติ 2 มิติ

# 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))

ผลลัพธ์

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]])

ตัวอย่างที่ 3

โปรแกรม Python ต่อไปนี้จะแสดงวิธีเปรียบเทียบเทนเซอร์ 1-D กับ 2-Dtensor องค์ประกอบ

# 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))

ผลลัพธ์

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]])