Reduction Operations

Operations that reduce arrays along specified dimensions.

Functions

• argmax
• max
• mean
• sum
• sum_batch_dims

Quick Reference

argmax

nabla.argmax(arg: 'Array', axes: 'int | None' = None, keep_dims: 'bool' = False) -> 'Array'

Nabla operation: argmax

max

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

Maximum of array elements over given axes.

mean

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

Arithmetic mean of array elements over given axes.

sum

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

Sum of array elements over given axes.

sum_batch_dims

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

Nabla operation: sum_batch_dims