Функция PyTorch torch.from_dlpack

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.from_dlpackЭто функция в PyTorch, используемая для создания тензора PyTorch из тензора dlpack.

Определение функции

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

Другие расширения