code: def _sum_rows(input: Tensor) -> Tensor:
      return input.reshape(input.shape[0], -1).sum(1)
  
imports: import typing
  from inspect import signature
  from typing import Any, Callable, List, Tuple, TYPE_CHECKING, Union
  import torch
  from captum._utils.common import _format_baseline, _format_output, _format_tensor_into_tuples, _validate_input
  from captum._utils.typing import BaselineType, Literal, TargetType, TensorOrTupleOfTensorsGeneric
  from captum.attr._utils.approximation_methods import SUPPORTED_METHODS
  from torch import Tensor
label_desc: Use image node_id 1 for calling a global function with example usage: _sum_rows(input) and returns: input