PyTorch torch.from_dlpack 函数
Pytorch torch 参考手册torch.from_dlpack 是 PyTorch 中用于从 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)
# 创建张量
x = torch.randn(3, 4)
# 获取 dlpack 胶囊
dlpack = x.toDlpack()
# 从 dlpack 创建新张量
y = torch.from_dlpack(dlpack)
print(y.shape)