Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để tính toán biểu đồ của tensor trong PyTorch?

Biểu đồ của tensor được tính bằng torch.histc () . Nó trả về một biểu đồ được biểu diễn dưới dạng tensor. Nó có bốn tham số: đầu vào, thùng, tối thiểu tối đa . Nó sắp xếp các phần tử thành các thùng có chiều rộng bằng nhau giữa min tối đa . Nó bỏ qua các phần tử nhỏ hơn min và lớn hơn tối đa .

Các bước

  • Nhập thư viện được yêu cầu. Trong tất cả các ví dụ Python sau, các thư viện Python bắt buộc là torch Matplotlib . Đảm bảo rằng bạn đã cài đặt chúng.

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

  • Tính toán torch.histc (input, bins =100, min =0, max =100) . Trả về hàng chục giá trị biểu đồ. Đặt các giá trị thùng, tối thiểu và tối đa phù hợp theo nhu cầu của bạn.

  • In biểu đồ được tính toán ở trên.

  • Hình dung biểu đồ dưới dạng biểu đồ thanh.

Ví dụ 1

 # Chương trình Python để tính toán biểu đồ của tensor # nhập các thư viện cần thiết nhập torchimport matplotlib.pyplot dưới dạng plt # Tạo tensorT =torch.Tensor ([2,3,1,2,3,4,3,2,3, 4,3,4]) print ("Original Tensor T:\ n", T) # Tính biểu đồ của tensorhist =torch.histc đã tạo ở trên (T, bins =5, min =0, max =4) print ( "Biểu đồ của T:\ n", lịch sử) 

Đầu ra

 Original Tensor T:tensor ([2., 3., 1., 2., 3., 4., 3., 2., 3., 4., 3., 4.]) Biểu đồ của T :tensor ([0., 1., 3., 5., 3.]) 

Ví dụ 2

 # Chương trình Python để tính toán biểu đồ của tensor # nhập các thư viện cần thiết nhập torchimport matplotlib.pyplot dưới dạng plt # Tạo tensorT =torch.Tensor ([2,3,1,2,3,4,3,2,3, 4,3,4]) print ("Original Tensor T:\ n", T) # Tính biểu đồ của tensorhist =torch.histc đã tạo ở trên (T, bins =5, min =0, max =4) # Visualize biểu đồ được tính toán ở trên dưới dạng biểu đồ thanh =5x =range (thùng) plt.bar (x, hist, align ='center') plt.xlabel ('Bins') plt.ylabel ('Frequency') plt.show ()  

Đầu ra

 Tensor gốc T:tensor ([2., 3., 1., 2., 3., 4., 3., 2., 3., 4., 3., 4.]) 

Làm thế nào để tính toán biểu đồ của tensor trong PyTorch?