Creating Pytorch Tensor Directly from GPU Memory

Introduction

DLPack

Implementation Details

Systems with CUDA

Systems based on Apple Silicon


This site uses Just the Docs, a documentation theme for Jekyll.