关于 Keras 入门 开发者指南 代码示例 Keras 3 API 文档 模型 API 层 API 回调 API 操作 API 优化器 指标 损失函数 数据加载 内置小型数据集 Keras 应用 混合精度 多设备分布 LayoutMap API DataParallel API ModelParallel API ModelParallel API 分布式工具 RNG API 实用工具 Keras 2 API 文档 KerasTuner:超参数调整 KerasHub:预训练模型
► Keras 3 API 文档 / 多设备分布

多设备分布

LayoutMap API

  • LayoutMap 类
  • DeviceMesh 类
  • TensorLayout 类
  • distribute_tensor 函数

DataParallel API

  • DataParallel 类

ModelParallel API

  • ModelParallel 类

ModelParallel API

  • ModelParallel 类

分布式工具

  • set_distribution 函数
  • distribution 函数
  • list_devices 函数
  • initialize 函数
多设备分布
LayoutMap API
DataParallel API
ModelParallel API
ModelParallel API
分布式工具
条款 | 隐私权政策