文档
注册
评分
提单
论坛
小AI

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词