Class NonZero
Represents a NonZero layer. This returns the indices of the elements of the input tensor that are not zero.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class NonZero : Layer
Constructors
NonZero(String, String)
Initializes and returns an instance of NonZero layer.
Declaration
public NonZero(string name, string input)
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. |
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 |