Class OneHot
Represents a OneHot
layer. This generates a one-hot tensor with a given depth
, indices
and values
.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class OneHot : Layer
Constructors
OneHot(String, String, String, String, Int32)
Initializes and returns an instance of OneHot
layer with a float value.
Declaration
public OneHot(string name, string indices, string depth, string values, int axis)
Parameters
Type | Name | Description |
---|---|---|
String | name | The name to use for the output tensor of the layer. |
String | indices | The name to use for the indices tensor of the layer. |
String | depth | The name to use for the scalar depth tensor of the layer. |
String | values | The name to use for the two-element off/on values tensor of the layer. |
Int32 | axis | The axis along which the layer adds the one-hot representation. |
Fields
axis
The axis along which the layer adds the one-hot representation.
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 |