PyTorch torch.from_dlpack 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.from_dlpackPyTorchにおいて、dlpackテンソルからPyTorchテンソルを作成するための関数です。

関数定義

torch.from_dlpack(ext_tensor)

使用例

例

import torch

# テンソルを作成
x = torch.randn(3, 4)

# dlpackカプセルを取得
dlpack = x.toDlpack()

# dlpackから新しいテンソルを作成
y = torch.from_dlpack(dlpack)

print(y.shape)

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張