จะรับประเภทข้อมูลของเทนเซอร์ใน PyTorch ได้อย่างไร

เมตริกซ์ของ PyTorch มีลักษณะเป็นเนื้อเดียวกัน กล่าวคือ องค์ประกอบทั้งหมดของเทนเซอร์มีประเภทข้อมูลเดียวกัน เราสามารถเข้าถึงประเภทข้อมูลของเทนเซอร์โดยใช้ ".dtype" คุณลักษณะของเทนเซอร์ ส่งกลับประเภทข้อมูลของเทนเซอร์

ขั้นตอน

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

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

• คำนวณ T.dtype . โดยที่ T คือเทนเซอร์ที่เราต้องการรับประเภทข้อมูล

• พิมพ์ประเภทข้อมูลของเทนเซอร์

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

โปรแกรม Python ต่อไปนี้แสดงวิธีรับประเภทข้อมูลของเทนเซอร์

# Import the library
import torch

# Create a tensor of random numbers of size 3x4
T = torch.randn(3,4)
print("Original Tensor T:\n", T)

# Get the data type of above tensor
data_type = T.dtype

# Print the data type of the tensor
print("Data type of tensor T:\n", data_type)

ผลลัพธ์

Original Tensor T:
tensor([[ 2.1768, -0.1328, 0.8155, -0.7967],
         [ 0.1194, 1.0465, 0.0779, 0.9103],
         [-0.1809, 1.8085, 0.8393, -0.2463]])
Data type of tensor T:
torch.float32

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

# Python program to get data type of a tensor
# Import the library
import torch

# Create a tensor of random numbers of size 3x4
T = torch.Tensor([1,2,3,4])
print("Original Tensor T:\n", T)

# Get the data type of above tensor
data_type = T.dtype

# Print the data type of the tensor
print("Data type of tensor T:\n", data_type)

ผลลัพธ์

Original Tensor T:
   tensor([1., 2., 3., 4.])
Data type of tensor T:
   torch.float32