unsqueeze function

VARP unsqueeze(
  1. VARP input, {
  2. List<int> axis = const [],
})

Implementation

VARP unsqueeze(VARP input, {List<int> axis = const []}) {
  final (axisPtr, axisSize) = axis.toNativeArrayI32();
  final rval = VARP.fromPointer(C.mnn_expr_Unsqueeze(input.ptr, axisPtr.cast(), axisSize));
  calloc.free(axisPtr);
  return rval;
}