ExpandDims

Inputs:

  • input: float32(NHWC)
  • dims: int32

Outputs:

  • output: float32

Backend:

    • CPU
    • Metal
    • Vulkan
    • OpenCL

Tensorflow op:

  • ExpandDims