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