Conv
功能
卷积。
输入
三个输入:
X:输入4D或5Dtensor。输入为4D时数据类型:float16、float;输入为5D时数据类型:float16。
W:权重tensor。
B:可选,偏差,一维tensor。
输出
一个输出:
Y:卷积输出tensor。
属性
auto_pad:可选,Feature Map为NCHW时支持NOTSET、SAME_UPPER、SAME_LOWER和VALID属性;Feature Map为NCDHW仅支持NOTSET属性。
dilations:输入4Dtensor时为4个整数的列表,5Dtensor时为5个整数的列表。指定用于扩张卷积的扩张率,H和W维度取值范围为[1, 255]。
group:从输入通道到输出通道的阻塞连接数,输入通道和输出通道都必须被“group”整除;数据类型为int32,必须设置为1。
pads:输入4Dtensor时为4个整数的列表,5Dtensor时为6个整数的列表。指定顶部、底部、左侧和右侧填充,取值范围为[0, 255]。
strides:输入4Dtensor时为4个整数的列表,5Dtensor时为5个整数的列表。指定沿高度H和宽度W的卷积步长。H和W维度取值范围为[1, 63],默认情况下,N和C尺寸设置为1。
约束
输入tensor,W维度取值范围为[1, 4096]。
权重tensor,H维度和W维度取值范围为[1, 255]。
不支持atc工具--precision_mode=must_keep_origin_dtype参数时输入类型为float和float64。
支持的ONNX版本
Opset v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单