Class ArgMax
Represents an ArgMax layer. This computes the indices of the maximum elements of the input tensor along a given axis.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class ArgMax : ArgReduceConstructors
ArgMax(String, String, Int32, Boolean, Boolean)
Initializes and returns an instance of ArgMax layer.
Declaration
public ArgMax(string name, string input, int axis, bool keepdims = true, bool selectLastIndex = false)Parameters
| Type | Name | Description |
|---|---|---|
| String | name | The name to use for the output tensor of the layer. |
| String | input | The name to use for the input tensor of the layer. |
| Int32 | axis | The axis along which to perform the operation. |
| Boolean | keepdims | Whether to keep the axis dimension in the output tensor. The default value is |
| Boolean | selectLastIndex | Whether to perform the operation from the back of the axis. The default value is |
Methods
Execute(Tensor[], ExecutionContext)
Executes the layer using the operations and variables from the ExecutionContext and returns the output tensor.
If the layer has more than one output, output tensors are saved to variables.
Declaration
public override Tensor Execute(Tensor[] inputTensors, ExecutionContext ctx)Parameters
| Type | Name | Description |
|---|---|---|
| Tensor[] | inputTensors | |
| ExecutionContext | ctx |
Returns
| Type | Description |
|---|---|
| Tensor |
Overrides
ToString()
Returns a string that represents the Layer.
Declaration
public override string ToString()Returns
| Type | Description |
|---|---|
| String |