Rate This Document
Findability
Accuracy
Completeness
Readability

Function Description

Operator Function

Performs a specified algorithm operation on each target element in one or more dimensions of a tensor. The formula is , where reduce_op includes the operations listed in Table 1 reduce_op algorithm operations.
Table 1 reduce_op algorithm operations

reduce_op

Function

max

Obtains the maximum value of a tensor along the reduction dimension.

min

Obtains the minimum value of a tensor along the reduction dimension.

sum

Obtains the sum of elements in a tensor along the reduction dimension.

mul

Obtains the product of elements in a tensor along the reduction dimension.

mean

Obtains the average value of elements in a tensor along the reduction dimension.