helper.make_node()是ONNX中的一个函数,用于创建一个节点(Node)。
该函数的参数包括:
name:节点的名称。
op_type:节点的操作类型。
inputs:节点的输入张量列表。
outputs:节点的输出张量列表。
attrs:节点的属性字典。
例如,以下代码创建了一个名为“Reshape”的节点,该节点将输入张量从形状(3,224,224)更改为形状(1,768):
reshape_node = helper.make_node('Reshape', inputs=['input'], outputs=['output'], shape=[1, 768])
节点是ONNX模型中的一个基本构建块,它表示一个操作或算子。每个节点都有一个类型(type),表示该节点执行的操作类型。
以下是一些常见的ONNX节点类型:
Add:加法运算
Div:除法运算
Mul:乘法运算
Sub:减法运算
Transpose:转置操作
ReduceMean:平均值归约操作
ReduceSum:求和归约操作
LogSoftmax:对数softmax归约操作
Softmax:softmax归约操作
Concat:连接操作
Split:分割操作
Pad:填充操作
Reshape:重塑操作
Conv:卷积操作
MaxPool:最大池化操作
AveragePool:平均池化操作
BatchNorm:批量归一化操作
LeakyReLU:泄漏整流线性单元激活函数
PRelu:部分整流线性单元激活函数
Elu:指数线性单元激活函数
Swish:双曲正切激活函数
Sigmoid:Sigmoid激活函数
Tanh:双曲正切激活函数
Identity:恒等节点,不执行任何操作。