Cách lấy kiểu dữ liệu của tensor trong PyTorch?

Tensor PyTorch là đồng nhất, nghĩa là tất cả phần tử của tensor có cùng kiểu dữ liệu. Chúng ta có thể truy cập kiểu dữ liệu của tensor bằng thuộc tính .dtype của tensor. Nó trả về kiểu dữ liệu của tensor.

Các bước

• Nhập thư viện cần thiết. Thư viện cần là torch. Đảm bảo đã cài đặt.

• Tạo tensor và in ra.

• Tính T.dtype. T là tensor muốn lấy kiểu dữ liệu.

• In kiểu dữ liệu của tensor.

Ví dụ 1

Chương trình Python sau minh họa cách lấy kiểu dữ liệu của tensor.

# Nhập thư viện
import torch

# Tạo tensor số ngẫu nhiên kích thước 3x4
T = torch.randn(3, 4)
print("Original Tensor T:\n", T)
# Lấy kiểu dữ liệu tensor
data_type = T.dtype
# In kiểu dữ liệu
print("Data type of tensor T:\n", data_type)

Kết quả

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

Ví dụ 2

# Chương trình Python lấy kiểu dữ liệu tensor
import torch

# Tạo tensor
T = torch.Tensor([1, 2, 3, 4])
print("Original Tensor T:\n", T)
# Lấy kiểu dữ liệu
data_type = T.dtype
# In kiểu dữ liệu
print("Data type of tensor T:\n", data_type)

Kết quả

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