module

reduction

Ops that accumulate or reduce a tensor along an axis.

Functions

• ​mean: Reduces a symbolic tensor using a mean operation.
• ​arg_max: Finds the index of the maximum value along a dimension.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!
If you'd like to share more information, please report an issue on GitHub

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other