代码拉取完成,页面将自动刷新
.. py:class:: mindspore.nn.MicroBatchInterleaved(network, interleave_num=2) 这个函数的作用是将输入在第零维度拆成 `interleave_num` 份,然后执行包裹的cell的计算。 使用场景:当在半自动模式以及网络中存在模型并行时,第1份的切片数据的前向计算同时,第2份的数据将会进行模型并行的通信,以此来达到通信计算并发的性能加速。 参数: - **network** (Cell) - 需要封装的网络。 - **interleave_num** (int,可选) - batch size的拆分份数,默认值: ``2`` 。 输入: tuple[Tensor],与传入的 `network` 的输入一致。 输出: 被封装后的网络。传入的 `network` 的输出只能是单个Tensor。
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。