Función PyTorch torch.from_dlpack

Pytorch torch 参考手册Manual de referencia de PyTorch: torch

torch.from_dlpackEs una función de PyTorch que se utiliza para crear un tensor de PyTorch a partir de un tensor dlpack.

Definición de la función

torch.from_dlpack(ext_tensor)

Ejemplo de uso

Ejemplo

import torch

# Crear tensor
x = torch.randn(3, 4)

# Obtener cápsula dlpack
dlpack = x.toDlpack()

# Crear nuevo tensor a partir de dlpack
y = torch.from_dlpack(dlpack)

print(y.shape)

Pytorch torch 参考手册Manual de referencia de PyTorch: torch

Otras extensiones