std::vector<int64_t> GetReduceAxes() const noexcept;

函数功能

获取进行归约的轴。

函数原型

std::vector<int64_t> GetReduceAxes() const noexcept;

参数说明

无

返回值说明

返回对应要进行归约操作的轴的集合。

例如：

• { 0 }, 表示要归约的轴为0轴。
• { 0, 1 }, 表示要归约的轴为0、1轴。