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

SpatialTransformer

输入

  • x:
    • 是否必填: 必填
    • 数据类型: fp16,fp32
    • 参数解释: NCHW
    • 规格限制: NCHW
  • theta:
    • 是否必填: 非必填
    • 数据类型: fp16,fp32
    • 参数解释:

      input_theta shape 是 (n, size),其中的size等于theta_1_1-theta_2_3中没有取值的数量,如theta_1_2=1,theta_2_1=2,theta_2_2=3,theta_2_3=4,input theta=[9,8],则最终的theta是: 9,1,8,2,3,4

    • 规格限制: ND

属性

  • output_H:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,则为x中shape的高
    • 规格限制: 无
  • output_W:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,则为x中shape的宽
    • 规格限制: 无
  • theta_1_1:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,表示由input中的theta提供
    • 规格限制: 无
  • theta_1_2:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,表示由input中的theta提供
    • 规格限制: 无
  • theta_1_3:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,表示由input中的theta提供
    • 规格限制: 无
  • theta_2_1:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,表示由input中的theta提供
    • 规格限制: 无
  • theta_2_2:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,表示由input中的theta提供
    • 规格限制: 无
  • theta_2_3:
    • 是否必填: 非必填
    • 数据类型: -
    • 参数解释: 如果不提供,表示由input中的theta提供
    • 规格限制: 无

输出

y:
  • 是否必填: 必填
  • 数据类型: fp16,fp32
  • 参数解释: NCHW
  • 规格限制: NCHW
搜索结果
找到“0”个结果

当前产品无相关内容

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