组件介绍
安装指南
快速开始
API参考
贡献指南
执行从特征维度切分到序列维度切分的 all-to-all 操作。
x (Tensor) - 特征维度切分的输入张量。
group (Union[CommGroup, CommGroupBase]) - 操作的通信组。
从特征维度切分到序列维度切分的张量。