GlobalAvgPool2D

@frozen
public struct GlobalAvgPool2D<Scalar> : ParameterlessLayer where Scalar : TensorFlowFloatingPoint

A global average pooling layer for spatial data.

• TangentVector

  Declaration

  public typealias TangentVector = EmptyTangentVector

• init()

  Creates a global average pooling layer.

  Declaration

  public init()

• forward(_:)

  Returns the output obtained from applying the layer to the given input.

  Declaration

  @differentiable
  public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>

  Parameters

  input

  The input to the layer.

  Return Value

  The output.