PyTorch torch.is_storage 関数
Pytorch torch リファレンスマニュアルtorch.is_storagePyTorchにおいて、オブジェクトがストレージオブジェクトであるかどうかを確認するために使用される関数です。
関数定義
torch.is_storage(obj)
使用例
実例
import torch
# テンソルを作成
x = torch.tensor([1, 2, 3])
# ストレージオブジェクトかどうかをチェック
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
# テンソルを作成
x = torch.tensor([1, 2, 3])
# ストレージオブジェクトかどうかをチェック
print(torch.is_storage(x))
print(torch.is_storage(x.storage()))
出力結果は次のとおりです：
False True
その他の拡張