Function _may_reduce_to_scalar

tensorflow/python/ops/math_ops.py:1454–1459

Set a reduction's output shape to be a scalar if we are certain.

(keepdims, axis, output)

def _may_reduce_to_scalar(keepdims, axis, output):
  """Set a reduction's output shape to be a scalar if we are certain."""
  if not common_shapes.has_fully_defined_shape(output) and (not keepdims) and (
      axis is None):
    output.set_shape(())
  return output


@tf_export(v1=["math.reduce_sum", "reduce_sum"])