Class Flatten
Represents a Flatten layer. The layer computes the output tensor by reshaping the input tensor into a 2D matrix according to the given axis.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class Flatten : Layer
Constructors
Flatten(String, String, Int32)
Initializes and returns an instance of Flatten layer.
Declaration
public Flatten(string name, string input, int axis = 1)
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. |
| Int32 | axis | The axis up to which to flatten the input dimensions into the first dimension of the output tensor. The default value is 1. |
Fields
axis
The axis up to which to flatten the input dimensions into the first dimension of the output tensor.
Declaration
public int axis
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 |