无
入门指南 开发者指南 代码示例 Keras 3 API 文档 模型 API 层 API 回调 API Ops API 优化器 指标 损失函数 数据加载 内置小型数据集 Keras 应用 混合精度 多设备分布 LayoutMap API DataParallel API ModelParallel API 分布工具 RNG API 重物化 工具 Keras 2 API 文档 KerasTuner: 超参数调优 KerasHub: 预训练模型 KerasRS
keras.io logo
  • 入门指南
  • 指南
  • API
  • 示例
  • Keras Tuner
  • Keras RS
  • Keras Hub
Keras 3 API 文档
模型 API 层 API 回调 API Ops API 优化器 指标 损失函数 数据加载 内置小型数据集 Keras 应用 混合精度 多设备分布 LayoutMap API DataParallel API ModelParallel API 分布工具 RNG API 重物化 工具
Keras 2 API 文档
► Keras 3 API 文档 / 多设备分布

多设备分布

LayoutMap API

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

DataParallel API

  • DataParallel 类

ModelParallel API

  • ModelParallel 类

分布工具

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