mx.nd.argmax.channel

Description

Returns argmax indices of each channel from the input array.

The result will be an NDArray of shape (num_channel,).

In case of multiple occurrences of the maximum values, the indices corresponding to the first occurrence are returned.

Example:

x = [[ 0.,  1.,  2.],
[ 3.,  4.,  5.]]

argmax_channel(x) = [ 2.,  2.]

Arguments

Argument

Description

data

NDArray-or-Symbol.

The input array