AvgPool2DOperator

public class AvgPool2DOperator: Pooling2DOperator

2D Average pooling.

Input shape

Input data should be a tensor with 3D shape [channel, height, width] or [height, width, channel] according to the channelPosition.


                    
                    
                    init(kernelSize:stride:channelPosition:paddingMode:inputTensors:outputTensors:computationDelegate:operatorLabel:)

Declaration

Swift

public required init(kernelSize: [Int], stride: [Int]? = nil,
                     channelPosition: TensorChannelOrder = TensorChannelOrder.First,
                     paddingMode: PaddingMode = PaddingMode.Valid,
                     inputTensors: [Tensor]? = nil,
                     outputTensors: [Tensor]? = nil,
                     computationDelegate: OperatorCalculationDelegate? = nil,
                     operatorLabel: String = "Pooling")

Show on GitHub