Method argMax

static member argMax: Tensor<'T> -> int64 list

Finds the indicies of the maximum value of the tensor.

Declaration
static member argMax: a:Tensor<'T> -> int64 list
Parameters
Type Name Description
Tensor<'T> a

The tensor containing the source values.
Returns
Type Description
int64 list

The indices of the position of the maximum value.
Remarks

The maximum value within the specified tensor is found and its indicies are returned. The function fails for an empty tensor.

Examples
 let a = HostTensor.ofList2D [[1.0; 2.0; 3.0; 4.0]
                              [5.0; 6.0; 7.0; 8.0]]
 let b = Tensor.argMax a // b = [1L; 3L]
See Also