Method ArgMax
ArgMax(TensorFloat, TensorInt, int, bool, bool)
Computes the indices of the maximum elements of the input tensor along a given axis.
Declaration
public override void ArgMax(TensorFloat X, TensorInt O, int axis, bool keepdim, bool selectLastIndex)Parameters
| Type | Name | Description |
|---|---|---|
| TensorFloat | X | The input tensor. |
| TensorInt | O | The output tensor to be computed and filled. |
| 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. |
Overrides
ArgMax(TensorInt, TensorInt, int, bool, bool)
Computes the indices of the maximum elements of the input tensor along a given axis.
Declaration
public override void ArgMax(TensorInt X, TensorInt O, int axis, bool keepdim, bool selectLastIndex)Parameters
| Type | Name | Description |
|---|---|---|
| TensorInt | X | The input tensor. |
| TensorInt | O | The output tensor to be computed and filled. |
| 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. |