AI常用算子列表
AI常用算子列表
Abs
AbsGrad
AcosGrad
Add
AsinGrad
AsinhGrad
AtanGrad
BiasAddGrad
BNTrainingUpdate
Ceil
ConcatD
Cos
Cosh
Div
Elu
EluGrad
Equal
Erf
Erfc
Exp
Expm1
Floor
Gelu
GeluGrad
Inv
InvGrad
L2Loss
L2Normalize
L2NormalizeGrad
Log
LogSoftmaxV2
LogSoftmaxGrad
Log1p
Maximum
Mod
Mul
Neg
OnesLike
Pow
PReluGrad
Reciprocal
ReciprocalGrad
ReduceAllD
ReduceAnyD
ReduceMaxD
ReduceMeanD
ReduceSumD
Relu
ReluGrad
Relu6
Relu6Grad
Rint
Round
Rsqrt
RsqrtGrad
Selu
Sigmoid
SigmoidCrossEntropyWithLogits
SigmoidGrad
Sign
Sinh
SmoothL1Loss
SoftmaxV2
SoftmaxGrad
Softplus
Softsign
SplitD
Sqrt
SqrtGrad
Square
StridedSliceD
Sub
Tanh
TanhGrad
SoftmaxCrossEntropyWithLogits
MatMul
GNTrainingReduce
GNTrainingUpdate
CosineEmbeddingLoss
INTrainingReduceV2
INTrainingUpdateV2
AvgPool
Conv2D
Conv2DBackpropInput
Conv2DBackpropFilter
Conv2DCompress
Conv2DTranspose
Conv3D
Conv3DBackpropInput
Conv3DBackpropFilter
DepthwiseConv2D
DepthwiseConv2DBackpropInput
DepthwiseConv2DBackpropFilter
Deconvolution
GEMM
Pooling
序号 |
算子 |
含义 |
1 |
Activation |
激活函数,支持模式:Abs、Sigmoid、Tanh、ELU、PReLU、Relu、Relu6 |
2 |
Add |
计算2个输入的和 |
3 |
Argmax |
返回输入的最大值对应的索引序号 |
4 |
BatchMatmul |
在Batch维度上,对切片进行Matmul运算 |
5 |
BatchNorm |
对输入在batch上做归一化计算 |
6 |
BatchToSpaceND |
将批处理数据重新排列到空间数据块中,然后进行裁剪 |
7 |
BiasAdd |
计算输入tensor和1个常量输入的和 |
8 |
BNInference |
对输入在batch上做归一化计算,同BatchNorm |
9 |
Cast |
转换输入的数据类型 |
10 |
Ceil |
向上取整 |
11 |
ClipByValue |
将一个张量中的数值限制在[min, max]范围之内 |
12 |
Concat |
数据按维度拼接 |
13 |
Convolution |
卷积 |
14 |
ConvolutionDepthwise |
深度卷积 |
15 |
ConvTranspose |
转置卷积,计算过程约等于卷积的反向计算 |
16 |
Cos |
计算cos |
17 |
Crop |
截取 |
18 |
CropAndResize |
从输入图像张量中提取作物并调整它们的大小 |
19 |
Deconvolution |
反卷积 |
20 |
DepthToSpace |
Channel维通过空间块转移至Height、Width维的一种数据重排 |
21 |
Eltwise |
按元素操作层(求和、乘积、最大值) |
22 |
Erf |
误差函数(也称为高斯误差函数) |
23 |
Equal |
判断第一个输入是否等于第二个输入 |
24 |
Exp |
计算并返回给定输入、给定参数的EXP值 |
25 |
ExpandDims |
对输入shape进行补维 |
26 |
Expm1 |
计算y = exp(x) - 1 |
27 |
Fill |
使用输入的维度和值信息,生成一个tensor |
28 |
Flatten |
保留输入的第一个维度,把第一个维度包含的每个子张量展开成一个行向量 |
29 |
FlattenV2 |
将输入张量扁平化为二维矩阵,可选起始和终止axis |
30 |
Floor |
向下取整 |
31 |
FloorDiv |
x / y 向下取整 |
32 |
FloorMod |
返回除法元素的余数 |
33 |
FullConnection |
全连接 |
34 |
Gather |
根据indices对输入tensor的第axis根轴获取合成新的tensor |
35 |
GatherNd |
将输入张量的切片聚合成具有indices指定维度的新张量 |
36 |
Gemm |
使用输入张量、一组学习的权重计算内积,并添加偏差 |
37 |
Greater |
判断第一个输入是否大于第二个输入 |
38 |
GreaterEqual |
按元素返回(x1 >= x2)真值 |
39 |
InstanceNorm |
对输入在像素上做归一化计算 |
40 |
Interp |
插值层 |
41 |
LeakyRelu |
激活函数 |
42 |
LayerNorm |
层归一化函数 |
43 |
Less |
按元素返回(x1 < x2)真值 |
44 |
LessEqual |
比较两个输入张量每个元素是否满足小于等于的关系 |
45 |
Log |
计算每个元素的自然对数 |
46 |
Log1p |
计算(1+x)元素的自然对数 |
47 |
LogicalAnd |
对两个输入张量做逻辑与运算 |
48 |
LogicalNot |
对输入张量做逻辑非运算 |
49 |
LogicalOr |
对两个输入张量做逻辑或运算 |
50 |
LRN |
局部响应标准化,用于防止数据过度拟合 |
51 |
Matmul |
对2个输入做矩阵乘法运算 |
52 |
Maximum |
计算2个输入中的较大值 |
53 |
MirrorPad |
根据指定的paddings来填充一个tensor |
54 |
Minimum |
返回两个输入中小的元素 |
55 |
Mul |
计算2个输入的乘积 |
56 |
Multinomial |
生成多项式分布的随机数据 |
57 |
Neg |
对每个元素值取反 |
58 |
NotEqual |
比较两个输入张量每个元素是否满足不等于的关系 |
59 |
NonMaxSuppressionV3D |
按分数降序选择边界框的子集 |
60 |
NonMaxSuppressionV6 |
选出与上一个框有较高的交叉点重叠率(IOU)的框 |
61 |
OneHot |
将input转化为独热类型数据输出 |
62 |
Pack |
将多个输入堆叠拼接成1个 |
63 |
Pad |
对输入做shape填充 |
64 |
PadV2 |
对输入做shape填充,填充值为constant_values |
65 |
Permute |
对输入做维度转置操作 |
66 |
Pooling |
池化层 |
67 |
Pow |
计算幂 |
68 |
Power |
计算y =(scale*x+shift)^ power |
69 |
Prelu |
执行参数ReLU,生成一个输出数据(Tensor) |
70 |
PriorBox |
在特征图的每个位置生成默认框 |
71 |
RealDiv |
根据实际数据类型,按元素返回x1/x2 |
72 |
ReducelogsumExp |
沿着axis指定的轴计算log(sum(exp(elements))) |
73 |
ReduceMax |
计算张量沿着某一维度的和,可以在求最大值后降维 |
74 |
ReduceMean |
计算一个张量中各维元素的均值 |
75 |
ReduceMin |
计算张量沿着某一维度的和,可以在求最小值后降维 |
76 |
ReduceProd |
计算一个张量各个维度上元素的乘积 |
77 |
ReduceSum |
计算张量沿着某一维度的和,可以在求和后降维 |
78 |
Reduction |
算张量沿着某一维度的和/绝对值和/平方和/均值和,降维运算 |
79 |
Reorg |
实现数据重排,调整tensor维度。以stride为步长对Channel、Height、Width放大或缩小 |
80 |
Reshape |
改变输入维度 |
81 |
ResizeBilinear |
将图片尺寸调整到指定的大小 |
82 |
ResizeNearestNeighbor |
使用最近邻差值算法进行图像大小调整 |
83 |
ResizeNearestNeighborV2 |
使用最近邻插值将图像大小调整为size大小 |
84 |
Rint |
查找并返回最接近“x”的整数,如果结果在两个可表示值之间选择偶数表示 |
85 |
Round |
将张量的值按元素四舍五入到最接近的整数 |
86 |
Rsqrt |
计算每个元素的平方根的倒数 |
87 |
Scale |
out=alpha*Input+beta |
88 |
ScatterUpdate |
将稀疏更新应用于变量引用 |
89 |
Select |
根据条件选择输出 |
90 |
ShuffleChannel |
对输入以Channel维度按照group做数据分组重排 |
91 |
ShuffleChannelV2 |
在axis维度做Shuffle运算 |
92 |
Sin |
计算sin |
93 |
Slice |
从输入张量中提取切片,此操作从由begin指定位置开始的张量input中提取一个尺寸size的切片 |
94 |
Softmax |
归一化逻辑函数 |
95 |
SpaceToBatchND |
重新排列的空间数据块成批 |
96 |
SpaceToDepth |
通过空间快将Height、Width维的数据转移至Channel维的一种数据重排 |
97 |
SparseToDense |
将稀疏表示转换为密集张量 |
98 |
Split |
沿指定维度将输入张量平均分割成一个张量列表 |
99 |
SplitV |
沿指定轴将输入张量分割为num_split个新的张量 |
100 |
Sqrt |
计算输入张量的平方根 |
101 |
Square |
计算每个元素的平方 |
102 |
Squeeze |
从输入张量的shape中删除一个值为1的维度 |
103 |
SSDDetectionOutput |
SSD网络的一个算子,它的作用是根据priorbox和其对应的偏移量、评分信息来生成检测目标的数量和候选框 |
104 |
StrideSlice |
对输入从起始位置到结束位置按照步长截取数据 |
105 |
StrideSliceV2 |
提取张量的一个分段切片 |
106 |
Sub |
计算2个输入的差 |
107 |
Tan |
计算Tan |
108 |
Threshold |
如果该值大于阈值,则输出为1;否则,输出为0 |
109 |
Tile |
平铺给定矩阵 |
110 |
TopK |
提取最后一个维度的k个最大值 |
111 |
TruncateDiv |
按元素返回x / y,对结果小数部分截断保留整数部分,截断表示负数将小数量向零舍入 |
112 |
Unpack |
Pack的逆操作,将数据沿axis轴进行分组,输出N个数据维度相同,维度数减1 |
113 |
Xlogy |
返回x乘以log(y) |
ONNX算子边界
序号 |
算子 |
含义 |
边界 |
1 |
Abs |
求绝对值 |
【输入】 input:输入Tensor,float32 【约束】 无限制 【输出】 output:输出Tensor |
2 |
Acos |
反余弦 |
【输入】 input:输入Tensor,数值范围[-1, 1],类型:float32 【约束】 无限制 【输出】 output:输出Tensor,数值范围[0, pi],类型与x输入相同 |
3 |
Add |
二元点加 |
【输入】
【约束】 支持两组输入的维度不一致,进行广播操作(广播即维度补齐),目前支持以下几种广播场景:
对于两个输入维度个数不相同的场景,需要将维度补齐到四维。例如,x.shape=(1, 5, 6, 7) 和 y.shape=(6, 7)需要将y的维度补齐到4维,即y.shape=(1, 1, 6, 7)。 说明:两个Tensor的输入顺序可以互换。 【输出】 C:输出Tensor,类型同B |
4 |
And |
取与运算 |
【输入】
【约束】 input输入不支持NCHW格式数据 【输出】 output:输出Tensor,类型:bool |
5 |
ArgMax |
在指定轴上找到最大值索引 |
【输入】 data:输入Tensor,类型:float32 【参数】
【输出】 output:输出Tensor,类型:int64 |
6 |
Asin |
反正弦 |
【输入】 input:输入Tensor,数值范围[-1, 1],类型:float32 【约束】 无限制 【输出】 output:输出Tensor,数值范围[-pi/2, pi/2],类型与input输入相同 |
7 |
Atan |
反正切 |
【输入】 input:输入Tensor,类型:float32 【约束】 无限制 【输出】 output:输出Tensor,类型与input输入相同 |
8 |
AveragePool |
平均池化 |
【输入】 X:输入Tensor,类型:float32 【参数】
【约束】
【输出】 Y:输出Tensor,类型与X输入相同 |
9 |
BatchNormalization |
对输入做标准化 |
【输入】
【参数】
【约束】
【输出】
|
10 |
Cast |
数据类型转换 |
【输入】 input:输入Tensor,类型:float32,bool,int32,uint8 【参数】 to:数据类型int
【约束】 支持下面类型转换:
【输出】 output:输出Tensor |
11 |
Ceil |
向上取值 |
【输入】 X:输入Tensor,类型:float32 【约束】 无 【输出】 output:输出Tensor,类型与input输入相同 |
12 |
Clip |
将输入限制在一个区间中 |
【输入】
【参数】 在opset 7~10中min和max是参数 【约束】
【输出】 output:输出Tensor,类型与input输入相同 |
13 |
Concat |
数据按维度拼接 |
【输入】
【参数】
【约束】 无 【输出】 concat_result:拼接之后的Tensor,类型:float32,int32 |
14 |
Constant |
输出1个常量Tensor |
【输入】 无 【参数】
【约束】
【输出】 output:输出Tensor,和提供的Tensor具有相同的值 |
15 |
ConstantOfShape |
根据给定的值和维度,生成1个Tensor |
【输入】 input:1-D Tensor,类型:int32,uint8 【参数】 value:optional,0-D Tensor,需要填充的值,默认为0,默认类型float32 【约束】 无 【输出】 output:输出Tensor,类型同value |
16 |
Conv |
卷积 |
【输入】
【参数】
【约束】 auto_pad参数不支持SAME_UPPER, SAME_LOWER 【输出】 Y:输出Tensor,类型同X |
17 |
ConvTranspose |
反卷积 |
【输入】
【参数】
【约束】
【输出】 Y:输出Tensor,类型同X输入 |
18 |
Cos |
计算余弦 |
【输入】 input:输入Tensor,类型:float32 【约束】 无限制 【输出】 output:输出Tensor,类型与x输入相同 |
19 |
DepthToSpace |
重组数据,根据blocksize |
【输入】 input:输入Tensor,类型:float32 【参数】
【约束】 mode仅支持DCR CRD两种模式 【输出】 output:输出Tensor,类型与x输入相同 |
20 |
Div |
做除法运算 |
【输入】
【约束】 对于两个输入维度个数不相同的场景,需要将维度补齐到四维。例如,x.shape=(1, 5, 6, 7) 和 y.shape=(6, 7)需要将y的维度补齐到4维,即y.shape=(1, 1, 6, 7)。 【输出】 C:输出Tensor,类型与x输入相同 |
21 |
Elu |
根据f(x) = alpha * (exp(x) - 1.) 该公式做计算 |
【输入】 X:输入Tensor,类型:float32 【参数】 alpha:float32,缺省值为1.0 【约束】 无 【输出】 Y:输出Tensor,类型与x输入相同 |
22 |
Equal |
判断输入是否相等 |
【输入】
【约束】 暂不支持广播场景 【输出】 C:输出Tensor,类型 bool |
23 |
Erf |
对输入数据逐个元素做error function计算 |
【输入】 x:输入Tensor,类型:float32 【约束】 无限制 【输出】 y:输出Tensor,类型与x输入相同 |
24 |
Exp |
指数函数,output = e^input |
【输入】 input:输入Tensor,类型:float32,double 【约束】 无限制 【输出】 output:输出Tensor,类型与x输入相同 |
25 |
Expand |
根据指定的shape做广播 |
【输入】
【约束】
【输出】 output:输出Tensor,类型与x输入相同 |
26 |
Flatten |
数据按维度展开 |
【输入】 input:输入 Tensor,类型float32 【参数】 axis:int,标识数据在哪个维度上展开,值的范围[-r, r],r是输入Tensor的维度个数,负值表示从最后1个维度往回计算 【约束】 axis参数必须为1 【输出】 output:2-D Tensor,类型float32 |
27 |
Floor |
对输入进行向下取整 |
【输入】 X:输入Tensor,类型:float32 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
28 |
Gather |
根据输入的indices,从data中获取entry组成输出tensor |
【输入】
【参数】 axis:int, [-r, r-1] ,r = rank(data) 【约束】 无 【输出】 output:输出Tensor,类型同data输入 |
29 |
Gemm |
通用矩阵乘法, Y = alpha * A' * B' + beta * C |
【输入】
【参数】
【约束】
【输出】 Y:输出Tensor,维度是(M, N) |
30 |
GlobalAveragePool |
对输入进行全局平均池化 |
【输入】 X:输入Tensor,类型:float32,维度是(N * C * H * W) 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
31 |
GlobalMaxPool |
对输入进行全局最大池化 |
【输入】 X:输入Tensor,类型float32,维度是(N * C * H * W) 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
32 |
Greater |
逐个元素比较哪个大 |
【输入】
【约束】 无 【输出】 C:输出Tensor,类型 bool |
33 |
InstanceNormalization |
按照下面公式做归一化运算 y = scale * (x - mean) / sqrt(variance + epsilon) + B |
【输入】
【参数】 alpha:float32,避免除零错误,默认值1e-05 【约束】 无 【输出】 output:输出Tensor,类型同input输入 |
34 |
LeakyRelu |
对输入进行LeakyRelu激活函数计算 f(x) = alpha * x for x < 0, f(x) = x for x >= 0 |
【输入】 X:输入Tensor,类型:float32 【参数】 alpha:float32,泄漏系数,默认值0.01 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
35 |
Less |
对输入A和B进行逐元素Less逻辑运算 |
【输入】
【约束】 无限制 【输出】 C:输出Tensor,类型:bool |
36 |
Log |
取自然对数运算 |
【输入】 input:输入Tensor,类型: float32 【约束】 无 【输出】 output:输出Tensor,类型同input输入 |
37 |
LogSoftmax |
对输入进行logsoftmax(log of softmax)计算 |
【输入】 input:输入Tensor,类型:float32 【参数】 axis:int,输入变为2D维度时的轴,默认值1 【约束】 axis参数仅支持最后一维 【输出】 output:输出Tensor,维度和输入相同 |
38 |
MatMul |
矩阵乘 |
【输入】
【约束】
【输出】 Y:输出Tensor,类型:float32 |
39 |
Max |
逐个元素取最大值 |
【输入】
【约束】
【输出】 y:输出Tensor,类型:float32 |
40 |
MaxPool |
最大池化 |
【输入】 X:输入Tensor,类型:float32 【参数】
【约束】
【输出】
|
41 |
Min |
逐个元素取最小 |
【输入】
【约束】
【输出】 y:输出Tensor,类型与Xn输入相同 |
42 |
Mul |
二元点乘 |
【输入】
【约束】 支持两组输入的维度不一致,进行广播操作(广播即维度补齐),目前支持以下几种广播场景:
对于两个输入维度个数不相同的场景,需要将维度补齐到四维。例如,x.shape=(1, 5, 6, 7) 和 y.shape=(6, 7)需要将y的维度补齐到4维,即y.shape=(1, 1, 6, 7)。 说明:两个Tensor的输入顺序可以互换。 【输出】 C:输出Tensor,类型同B |
43 |
Neg |
对Tensor的每个元素取反,y=-x |
【输入】 X:输入Tensor,类型:float32 【约束】 无 【输出】 Y:输出Tensor,类型:float32 |
44 |
NonMaxSuppression |
进行非最大值压缩在指定框中 |
【输入】
【约束】
【输出】 selected_indices:输出Tensor,类型:float32 |
45 |
Or |
对输入Tensor的每个元素取逻辑或 |
【输入】
【约束】 input输入不支持NCHW格式数据 【输出】 C:输出Tensor,类型:bool |
46 |
PRelu |
根据下面公式做运算 f(x) = slope * x for x < 0, f(x) = x for x >= 0 |
【输入】
【约束】
【输出】 Y:输出Tensor,类型同X输入 |
47 |
Pad |
对输入Tensor做补pad处理 |
【输入】
【参数】
【约束】
【输出】 output:输出Tensor,类型:float32 |
48 |
Pow |
逐个元素做指数运算 |
【输入】
【约束】 无限制 【输出】 Z:输出Tensor,类型:float32, int32 |
49 |
Range |
创建一个Tensor,Tensor的数据以start开始,以delta作为步长扩展直到limit |
【输入】
【约束】
【输出】 output:输出Tensor,标量,类型float32,int32 |
50 |
ReduceLogSumExp |
计算输入Tensor沿着指定轴上的对数和的指数 |
【输入】 data:输入Tensor,类型:float32 【参数】
【约束】 无 【输出】 reduced:输出Tensor |
51 |
ReduceMax |
计算输入Tensor沿着指定轴上的最大值 |
【输入】 data:输入Tensor,类型:float32 【参数】
【约束】 axes参数为必选参数 【输出】 reduced:输出Tensor |
52 |
ReduceMean |
计算输入Tensor沿着指定轴上的平均值 |
【输入】 data:输入Tensor,类型:float32 【参数】
【约束】 axes参数为必选参数 【输出】 reduced:输出Tensor |
53 |
ReduceMin |
计算输入Tensor沿着指定轴上的最小值 |
【输入】 data:输入Tensor,类型:float32 【参数】
【约束】
【输出】 reduced:输出Tensor |
54 |
ReduceSum |
计算输入Tensor沿着指定轴上的和 |
【输入】 data:输入Tensor,类型:float32 【参数】
【约束】 axes参数为必选参数 【输出】 reduced:输出Tensor |
55 |
Relu |
整流线性单位函数 |
【输入】 X:输入Tensor,类型:float32 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
56 |
Reshape |
改变输入Tensor的维度 |
【输入】
【约束】 shape输入必须是常量 【输出】 reshaped:输出Tensor,类型同data输入 |
57 |
Resize |
调整输入tensor的维度 |
【输入】
【参数】
【约束】
【输出】 Y:输出Tensor,类型同data输入 |
58 |
RoiAlign |
对关注的区域做对齐操作 |
【输入】
【参数】
【约束】 mode仅支持avg模式 【输出】 Y:输出Tensor,类型同X输入 |
59 |
Round |
逐个元素取整 |
【输入】 X:输入Tensor,类型:float32 【约束】 无限制 【输出】 Y:输出Tensor,类型同X输入 |
60 |
ScatterElements |
绘制散点图 |
【输入】
【参数】 axis:int,缺省值为0 【约束】 无 【输出】 output:输出Tensor,类型同data输入 |
61 |
Selu |
对输入做下面公式计算 y = gamma * (alpha * e^x - alpha) for x <= 0, y = gamma * x for x > 0 |
【输入】 x:输入Tensor,类型:float32 【参数】
【约束】 仅支持alpha和gamma参数是默认值 【输出】 y:输出Tensor,类型同x输入 |
62 |
Shape |
获取输入Tensor的维度,并输出 |
【输入】 data:输入Tensor,类型:float32,int32,bool,uint8 【约束】 不支持指定为输出节点 【输出】 shape:输出Tensor,data输入的维度 |
63 |
Sigmoid |
sigmoid函数,y = 1 / (1 + exp(-x)) |
【输入】 X:输入Tensor,类型:float32 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
64 |
Sign |
sign函数(符号函数), 当x>0,sign(x)=1 当x=0,sign(x)=0 当x<0,sign(x)=-1 |
【输入】 input:输入Tensor,类型:float32 【约束】 无 【输出】 output:输出Tensor,类型和维度同input输入 |
65 |
Sin |
计算正弦 |
【输入】 input:输入Tensor,类型:float32 【约束】 无限制 【输出】 output:输出Tensor,类型与x输入相同 |
66 |
Slice |
对输入Tensor沿着指定轴切分 |
【输入】
【参数】 opset9及之前版本,axes,ends,starts是参数 【约束】
【输出】 output:输出Tensor,切分后的Tensor |
67 |
Softmax |
归一化逻辑函数 |
【输入】 input:输入Tensor,类型:float32 【参数】 axis:int,输入变为2D维度时的轴,默认值1 【约束】 axis参数仅支持最后一维 【输出】 output:输出Tensor,类型和维度同input输入 |
68 |
Softplus |
softplus激活函数,y = ln(exp(x) + 1) |
【输入】 X:1-D输入Tensor,类型:float32 【约束】 无 【输出】 Y:1-D输出Tensor,类型同X输入 |
69 |
Split |
在指定轴上做拆分,输出多个tensor |
【输入】 input:输入Tensor,类型: float32 【参数】
【约束】 无 【输出】 outputs:D输出Tensor,类型同输入 |
70 |
Sqrt |
求平方根 |
【输入】 X:输入Tensor,类型:float32 【约束】 无 【输出】 Y:输出Tensor,类型同X输入 |
71 |
Squeeze |
在指定轴上降维 |
【输入】 data:输入Tensor,类型:float32 【参数】 axes:list of ints,轴,缺省值为所有维度,负值表示从后往前遍历 【约束】 无 【输出】 squeezed:输出Tensor,类型:float32 |
72 |
Sub |
逐个元素的减法运算 |
【输入】
【约束】 无 【输出】 C:输出Tensor,类型:float32 |
73 |
Tan |
正切函数 |
【输入】 input:输入Tensor,类型:float32 【约束】 无 【输出】 output:输出Tensor,类型:float32 |
74 |
Tanh |
双曲函数 |
【输入】 input:输入Tensor,类型:float32 【约束】 无 【输出】 output:输出Tensor,类型:float32 |
75 |
Tile |
对输入Tensor做平铺操作 |
【输入】
【约束】 repeats输入必须是常量 【输出】 output:输出Tensor,类型同input输入 |
76 |
TopK |
实现最大或者最小的K个元素在指定的轴上。 |
【输入】
【参数】
【约束】 K输入必须是常量 【输出】
|
77 |
Transpose |
根据属性perm中各个轴的排列顺序,对输入Tensor和shape做相应转换 |
【输入】 data:输入Tensor,类型:float32 【参数】 perm:list of ints,轴调整排列的顺序表 【约束】 无 【输出】 transposed:输出Tensor,类型:float32 |
78 |
Unsqueeze |
在指定轴上扩维 |
【输入】 data:输入Tensor,类型:float32,int32,uint8,bool 【参数】 axes:list of ints,指定需要扩维的轴 【约束】 仅支持axes个数为1 【输出】 expanded:输出Tensor,类型:float32,int32,uint8,bool |
79 |
Upsample |
上采样 |
【输入】
【参数】
【约束】
【输出】 Y:输出Tensor,类型:float32 |
80 |
ReduceL2 (V510新增) |
计算输入Tensor沿着指定轴上的欧几里得范数 |
【输入】 data:输入Tensor,类型:float32 【参数】
【约束】
【输出】 reduced:输出Tensor |