PyTorch torch.from_dlpack Funktion

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

torch.from_dlpackEs handelt sich um eine Funktion in PyTorch, die verwendet wird, um einen PyTorch-Tensor aus einem dlpack-Tensor zu erstellen.

Funktionsdefinition

torch.from_dlpack(ext_tensor)

Anwendungsbeispiel

Beispiel

import torch

# Tensor erstellen
x = torch.randn(3, 4)

# dlpack-Kapsel abrufen
dlpack = x.toDlpack()

# Einen neuen Tensor aus dlpack erstellen
y = torch.from_dlpack(dlpack)

print(y.shape)

Pytorch torch 参考手册PyTorch torch Referenzhandbuch

Andere Erweiterungen