鲲鹏社区首页
中文
注册
开发者
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

功能描述

算子功能

Shuffle算子可以对tensor以某一维度为轴心进行数据混洗。数学公式表达为:

其中

c′和c存在如下等式关系:

式子中

表1 公式参数说明

参数

参数说明

src

源输入Tensor。

dst

目标Tensor。

c

c维度为shuffle轴。

最外层索引。

最内层索引。