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 |