Class GatherND
Represents a GatherND layer. This takes slices of values from the batched input tensor indexed by the indices tensor.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class GatherND : LayerConstructors
GatherND(String, String, String, Int32)
Initializes and returns an instance of GatherND layer.
Declaration
public GatherND(string name, string input, string indices, int batchDims)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. |
| String | indices | The name to use for the indices tensor of the layer. |
| Int32 | batchDims | The number of batch dimensions of the input tensor, the gather begins at the next dimension. |
Fields
batchDims
The number of batch dimensions of the input tensor. The gather begins at the next dimension.
Declaration
public int batchDimsField 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 |