Class Softmax

Represents a Softmax activation layer along an axis: f(x, axis) = exp(X) / ReduceSum(exp(X), axis).

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

Constructors

Softmax(String, String, Int32)

Initializes and returns an instance of Softmax activation layer.

Declaration
public Softmax(string name, string input, int axis = -1)
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 apply the Softmax activation function.

Fields

axis

The axis along which to apply the Softmax activation function.

Declaration
public int axis
Field Value
Type Description
Int32

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