Class Split
Represents a Split
layer. The layer computes the output tensors by splitting the input tensor along a single given axis.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class Split : Layer
Constructors
Split(String, String, String, String[], Int32)
Initializes and returns an instance of Split
layer where the input tensor is split according to the split tensor.
Declaration
public Split(string name, string input, string split, string[] outputs, 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 | split | The name to use for the 1D split tensor of the layer. |
String[] | outputs | The names to use for all of the output tensors of the layer. |
Int32 | axis | The axis along which to split. |
Split(String, String, String[], Int32, Int32)
Initializes and returns an instance of Split
layer where the input tensor is split equally.
Declaration
public Split(string name, string input, string[] outputs, int axis, int numOutputs)
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[] | outputs | The names to use for all of the output tensors of the layer. |
Int32 | axis | The axis along which to split. |
Int32 | numOutputs | The number of outputs to split the input tensor into. |
Fields
axis
The axis along which to split.
Declaration
public int axis
Field Value
Type | Description |
---|---|
Int32 |
numOutputs
The number of outputs along which to split the input tensor if no split tensor is used.
Declaration
public int numOutputs
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 |