onnx创建节点

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:恒等节点,不执行任何操作。

猜你喜欢

转载自blog.csdn.net/qq_16792139/article/details/131431639