SpatialTransformer

Input

• x:
  • Required: yes
  • Type: fp16, fp32
  • Description: NCHW
  • Restriction: NCHW
• theta:
  • Required: no
  • Type: fp16, fp32
  • Description:

    The input theta has the shape (n, size), where size is equal to the number of from theta_1_1 to theta_2_3. For example, if theta_1_2=1, theta_2_1=2, theta_2_2=3, theta_2_3=4, input theta=[9,8], the final theta is 9,1,8,2,3,4.

  • Restriction: ND

Attribute

• output_H:
  • Required: no
  • Type: -
  • Description: If it is not specified, the height in the shape of x is used.
  • Restrictions: none

• output_W:
  • Required: no
  • Type: -
  • Description: If it is not specified, the width in the shape of x is used.
  • Restrictions: none

Output