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

ArgReduce

ArgMax

Inherited Members

ArgReduce.axis

ArgReduce.keepdims

ArgReduce.selectLastIndex

Layer.inputs

Layer.name

Layer.outputs

Layer.flags

object.Equals(object)

object.Equals(object, object)

object.ReferenceEquals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

Namespace: Unity.Sentis.Layers

Assembly: solution.dll

Syntax

[Serializable]
public class ArgMax : ArgReduce

Constructors

Name	Description
ArgMax(string, string, int, bool, bool)	Initializes and returns an instance of ArgMax layer.

Methods

Name	Description
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.
ToString()	Returns a string that represents the Layer.