Function ExpandDimIf0D

oneflow/core/common/shape.cpp:123–126

}

Shape ExpandDimIf0D(const Shape& shape) {
  if (shape.NumAxes() == 0) { return {1}; }
  return shape;
}

Shape ExpandDimIf0D(ShapeView shape) {
  if (shape.NumAxes() == 0) { return {1}; }