Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

GlobalMaxPool1D

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

A global max pooling layer for temporal data.

  • Declaration

    public typealias TangentVector = EmptyTangentVector

  • Creates a global max pooling layer.

    Declaration

    public init()

  • 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.
    context

    The contextual information for the layer application, e.g. the current learning phase.

    Return Value

    The output.