TensorPlay AI

教程

从第一个 CNN 开始,逐步进入自动微分、静态图与 C++ 扩展。

基础

进阶

  • 自定义数据集与 transforms
  • 保存、加载模型与 state_dict

高级

  • 使用 Stax 捕获静态图并实验算子融合。
  • 使用 p10::TensorImpl、Dispatcher 与 nanobind 构建 C++ 或 CUDA 扩展。
Ask DeepWiki