PyTorchでテンソルのヒストグラムを計算する方法は？

テンソルのヒストグラムは、 torch.histc（）を使用して計算されます 。テンソルとして表されるヒストグラムを返します。 input、bins、minの4つのパラメータを取ります および最大 。要素を分の間の等しい幅のビンに並べ替えます および最大 。 分よりも小さい要素は無視されます 最大よりも大きい 。

ステップ

• 必要なライブラリをインポートします。以下のすべてのPythonの例では、必要なPythonライブラリはトーチです。 およびMatplotlib 。すでにインストールされていることを確認してください。

• テンソルを作成して印刷します。

• torch.histc（input、bins =100、min =0、max =100）を計算します 。ヒストグラム値のテンソルを返します。必要に応じて、ビン、最小、最大を適切な値に設定します。

• 上記で計算したヒストグラムを印刷します。

• ヒストグラムを棒グラフとして視覚化します。

例1

# Python program to calculate histogram of a tensor
# import necessary libraries
import torch
import matplotlib.pyplot as plt

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

# Calculate the histogram of the above created tensor
hist = torch.histc(T, bins = 5, min = 0, max = 4)
print("Histogram of T:\n", hist)

出力

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

例2

# Python program to calculate histogram of a tensor
# import necessary libraries
import torch
import matplotlib.pyplot as plt

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

# Calculate the histogram of the above created tensor
hist = torch.histc(T, bins = 5, min = 0, max = 4)

# Visualize above calculated histogram as bar diagram
bins = 5
x = range(bins)
plt.bar(x, hist, align='center')
plt.xlabel('Bins')
plt.ylabel('Frequency')
plt.show()

出力

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