shape function

VARP shape(
  1. VARP input, {
  2. bool nchw = false,
})

Returns the shape of a variable.

Args:

  • input: A variable.

Returns:

  • A variable of Halide_Type_Int.

Implementation

VARP shape(VARP input, {bool nchw = false}) => VARP.fromPointer(C.mnn_expr_Shape(input.ptr, nchw));