sum_batch_dims

Signature

nabla.sum_batch_dims(arg: 'Array', axes: 'int | list[int] | tuple[int, ...] | None' = None, keep_dims: 'bool' = False) -> 'Array'

Description

sum array elements over given batch dimension axes.