Class ArgMax

Represents an ArgMax layer. This computes the indices of the maximum elements of the input tensor along a given axis.

Inheritance
Object
Layer
ArgMax
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class ArgMax : Layer

Constructors

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 true.
Boolean selectLastIndex

Whether to perform the operation from the back of the axis. The default value is false.

Fields

axis

The axis along which to perform the operation.

Declaration
public int axis
Field Value
Type Description
Int32

keepdims

Whether to keep the axis dimension in the output tensor.

Declaration
public bool keepdims
Field Value
Type Description
Boolean

selectLastIndex

Whether to perform the operation from the back of the axis.

Declaration
public bool selectLastIndex
Field Value
Type Description
Boolean

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
Overrides