Class AveragePool
Represents an AveragePool
pooling layer. This calculates an output tensor by pooling the mean values of the input tensor across its spatial dimensions according to the given pool and stride values.
Inherited Members
Namespace: Unity.Sentis.Layers
Syntax
[Serializable]
public class AveragePool : LocalPool
Constructors
AveragePool(String, String, Int32[], Int32[], Int32[], AutoPad)
Initializes and returns an instance of AveragePool
pooling layer.
Declaration
public AveragePool(string name, string input, int[] pool, int[] stride, int[] pad, AutoPad autopad = AutoPad.NotSet)
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[] | pool | The size of the kernel along each spatial axis. |
Int32[] | stride | The stride along each spatial axis. |
Int32[] | pad | The lower and upper padding values for each spatial dimension, [pad_left, pad_right] for 1D, [pad_top, pad_bottom, pad_left, pad_right] for 2D, etc. |
AutoPad | autopad | The auto padding mode of the pool as an |
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 |