input_tensor (np.ndarray or Tensor or list[Tensor]) – The tensor whose elements you want to sum. If you pass a list of tensors, they must all have
the same shape, and are interpreted as being stacked along a new first dimension (for
example, if you pass two 2D tensors of shape
[3, 4], the result is equivalent to passing
the stacked 3D tensor of shape
[2, 3, 4]).
axis (int or list[int] or tuple[int], optional) – The dimension or dimensions along which you want to sum the tensor. Defaults to None, in
which case this node sums along all axes of the tensor.
keepdims (bool, optional) – Whether or not to retain summed axes in the output tensor. If true, each dimension in
axis has size 1 in the result; otherwise, the dimensions in axis are removed from the
result. Defaults to false.
name (str, optional) – The name of the node.