Class ArgMin
Represents an ArgMin layer. This computes the indices of the minimum elements of the input tensor along a given axis.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class ArgMin : ArgReduceConstructors
ArgMin(String, String, Int32, Boolean, Boolean)
Initializes and returns an instance of ArgMin layer.
Declaration
public ArgMin(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 |