PyTorch torch.as_tensor 関数
Pytorch torch リファレンスマニュアルtorch.as_tensorは PyTorch においてデータをテンソルに変換するための関数です。torch.tensorとは異なり、可能な限りデータメモリを共有します。
関数定義
torch.as_tensor(data, dtype=None, device=None)
使用例
例
import torch
import numpy as np
# NumPy 配列から作成（メモリ共有）
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# NumPy 配列を変更
numpy_array[0] = 100
print("テンソル:", x) # テンソルも変更されます
import numpy as np
# NumPy 配列から作成（メモリ共有）
numpy_array = np.array([1, 2, 3])
x = torch.as_tensor(numpy_array)
# NumPy 配列を変更
numpy_array[0] = 100
print("テンソル:", x) # テンソルも変更されます
出力結果は以下のとおりです。
张量: tensor([100, 2, 3])
その他の拡張