AffineGrid
功能
给定一批矩阵,生成采样网格
边界
【输入】
俩个输入
theta:一个tensor,数据类型:float16、float32
output_size:一个tensor,数据类型:int32
【属性】
一个属性:
align_corners:bool型
【输出】
一个输出
y:一个tensor,数据类型:int
支持的ONNX版本
自定义算子,无对应onnx版本
父主题: 支持ONNX算子清单
给定一批矩阵,生成采样网格
【输入】
俩个输入
theta:一个tensor,数据类型:float16、float32
output_size:一个tensor,数据类型:int32
【属性】
一个属性:
align_corners:bool型
【输出】
一个输出
y:一个tensor,数据类型:int
自定义算子,无对应onnx版本