Fonction PyTorch torch.from_dlpack

Pytorch torch 参考手册 Manuel de référence Pytorch torch

torch.from_dlpackC'est une fonction de PyTorch utilisée pour créer un tenseur PyTorch à partir d'un tenseur dlpack.

Définition de fonction

torch.from_dlpack(ext_tensor)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur
x = torch.randn(3, 4)

# Obtenir la capsule dlpack
dlpack = x.toDlpack()

# Créer un nouveau tenseur à partir de dlpack
y = torch.from_dlpack(dlpack)

print(y.shape)

Pytorch torch 参考手册 Manuel de référence Pytorch torch

Autres extensions