Expand description
Tensor 后端抽象:支持多种 backend 实现
本模块提供了 tensor 后端的抽象层次,支持:
- NdArrayBackend: 基于 ndarray 的 CPU backend
- DfdxBackend: 基于 dfdx 的 GPU backend(支持自动微分)
- CandleBackend: 基于 candle 的轻量级 backend
Structs§
- NdArray
Storage - NdArray 存储后端
Enums§
- Unified
Storage - 统一的存储后端枚举
Traits§
- Tensor
Storage - Tensor 存储后端 trait