Split

概述

算子沿 axis 维度将x拆分成多个张量,张量数量由outputNum指定。

该算子对应的NodeType为NODE_TYPE_SPLIT。

输入:

  • x,n维tensor

输出:

  • outputs,一组n维张量,每一个张量类型和维度相同,每个张量的类型和x一致。

Since:

3.2

Version:

1.0

相关模块:

NNRt

汇总

Public 属性

名称 描述
outputNum long
指定分割数量。
sizeSplits long[]
指定输入的张量沿axis轴拆分后,每个张量的大小。
axis long
指定分割的维度。

类成员变量说明

axis

long Split::axis

描述:

指定分割的维度。

outputNum

long Split::outputNum

描述:

指定分割数量。

sizeSplits

long [] Split::sizeSplits

描述:

指定输入的张量沿 axis 轴拆分后,每个张量的大小。

如果 sizeSplits 的数据为空,则 sizeSplits 被拆分成大小均等的 张量,此时要求 x.shape[axis] 可以被 outputNum 整除;如果 sizeSplits 不为空,则要求 sizeSplits 所有元素之和等于 x.shape[axis]。