public struct GlobalAvgPool2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

A global average pooling layer for spatial data.

  • Creates a global average pooling layer.

    Declaration

    public init()
  • Returns the output obtained from applying the layer to the given input.

    Declaration

    public func call(_ input: Tensor<Scalar>) -> Tensor<Scalar>

    Parameters

    input

    The input to the layer.

    Return Value

    The output.