Skip to main content

Module backend

Module backend 

Source
Expand description

Tensor 后端抽象:支持多种 backend 实现

本模块提供了 tensor 后端的抽象层次,支持:

  • NdArrayBackend: 基于 ndarray 的 CPU backend
  • DfdxBackend: 基于 dfdx 的 GPU backend(支持自动微分)
  • CandleBackend: 基于 candle 的轻量级 backend

Structs§

NdArrayStorage
NdArray 存储后端

Enums§

UnifiedStorage
统一的存储后端枚举

Traits§

TensorStorage
Tensor 存储后端 trait