Class GatherElements
Represents a GatherElements layer. This takes values from the input tensor indexed by the indices tensor along a given axis.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class GatherElements : LayerConstructors
GatherElements(String, String, String, Int32)
Initializes and returns an instance of GatherElements layer.
Declaration
public GatherElements(string name, string input, string indices, int axis)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 | axis | The axis along which to perform the gather. |
Fields
axis
The axis along which to perform the gather.
Declaration
public int axisField 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 |