Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách tính histogram của tensor trong PyTorch?

Histogram của tensor được tính bằng torch.histc(). Nó trả về histogram biểu diễn dưới dạng tensor. Nó nhận bốn tham số: input, bins, min và max. Nó phân loại các phần tử vào các bin cùng độ rộng giữa min và max. Nó bỏ qua phần tử nhỏ hơn min và lớn hơn max.

Các bước

  • Nhập thư viện torch và Matplotlib. Đảm bảo đã cài.

  • Tạo tensor và in ra.

  • Tính torch.histc(input, bins=100, min=0, max=100). Trả về tensor giá trị histogram. Đặt bins, min, max phù hợp.

  • In histogram đã tính.

  • Trực quan histogram bằng biểu đồ cột.

Ví dụ 1

# Chương trình tính histogram tensor
import torch
import matplotlib.pyplot as plt

# Tạo tensor
T = torch.Tensor([2, 3, 1, 2, 3, 4, 3, 2, 3, 4, 3, 4])
print("Original Tensor T:\n", T)
# Tính histogram
hist = torch.histc(T, bins=5, min=0, max=4)
print("Histogram of T:\n", hist)

Kết quả

Original Tensor T:
tensor([2., 3., 1., 2., 3., 4., 3., 2., 3., 4., 3., 4.])
Histogram of T:
tensor([0., 1., 3., 5., 3.])

Ví dụ 2

# Chương trình tính histogram tensor
import torch
import matplotlib.pyplot as plt

# Tạo tensor
T = torch.Tensor([2, 3, 1, 2, 3, 4, 3, 2, 3, 4, 3, 4])
print("Original Tensor T:\n", T)
# Tính histogram
hist = torch.histc(T, bins=5, min=0, max=4)
# Trực quan histogram
bins = 5
x = range(bins)
plt.bar(x, hist, align='center')
plt.xlabel('Bins')
plt.ylabel('Frequency')
plt.show()

Kết quả

Original Tensor T:
tensor([2., 3., 1., 2., 3., 4., 3., 2., 3., 4., 3., 4.])

Cách tính histogram của tensor trong PyTorch?