Method ArgMax
ArgMax(TensorFloat, int, bool, bool)
Computes the indices of the maximum elements of the input tensor along a given axis.
Declaration
public TensorInt ArgMax(TensorFloat X, int axis, bool keepdim, bool selectLastIndex = false)Parameters
| Type | Name | Description |
|---|---|---|
| TensorFloat | X | The input tensor. |
| int | axis | The axis along which to reduce. |
| bool | keepdim | Whether to keep the reduced axes in the output tensor. |
| bool | selectLastIndex | Whether to perform the operation from the back of the axis. |
Returns
| Type | Description |
|---|---|
| TensorInt | The computed output tensor. |
ArgMax(TensorInt, int, bool, bool)
Computes the indices of the maximum elements of the input tensor along a given axis.
Declaration
public TensorInt ArgMax(TensorInt X, int axis, bool keepdim, bool selectLastIndex = false)Parameters
| Type | Name | Description |
|---|---|---|
| TensorInt | X | The input tensor. |
| int | axis | The axis along which to reduce. |
| bool | keepdim | Whether to keep the reduced axes in the output tensor. |
| bool | selectLastIndex | Whether to perform the operation from the back of the axis. |
Returns
| Type | Description |
|---|---|
| TensorInt | The computed output tensor. |