Class Squeeze
Represents a Squeeze
layer. The layer computes the output tensor by reshaping the input tensor by removing dimensions of size 1.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class Squeeze : Layer
Constructors
Squeeze(String, String)
Initializes and returns an instance of Squeeze
layer where the layer squeezes all the axes of size 1 from the input.
Declaration
public Squeeze(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. |
Squeeze(String, String, String)
Initializes and returns an instance of Squeeze
layer where the layer squeezes the specified axes of size 1 from the input.
Declaration
public Squeeze(string name, string input, string axes)
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 | axes | The name to use for the 1D axes 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 |