AvgPool3D

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

Una capa de agrupación promedio para datos espaciales o espacio-temporales.

  • Declaración

    public typealias TangentVector = EmptyTangentVector
  • El tamaño de la ventana de reducción deslizante para pooling.

    Declaración

    @noDerivative
    public let poolSize: (Int, Int, Int, Int, Int)
  • Los pasos de la ventana deslizante para cada dimensión de una entrada 5-D. Los pasos en dimensiones no espaciales deben ser 1 .

    Declaración

    @noDerivative
    public let strides: (Int, Int, Int, Int, Int)
  • El algoritmo de relleno para la agrupación.

    Declaración

    @noDerivative
    public let padding: Padding
  • Crea una capa de agrupación promedio.

    Declaración

    public init(
      poolSize: (Int, Int, Int, Int, Int),
      strides: (Int, Int, Int, Int, Int),
      padding: Padding
    )
  • Devuelve el resultado obtenido al aplicar la capa a la entrada dada.

    Declaración

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

    Parámetros

    input

    La entrada a la capa.

    Valor de retorno

    La salida.

  • Crea una capa de agrupación promedio.

    Declaración

    public init(poolSize: (Int, Int, Int), strides: (Int, Int, Int), padding: Padding = .valid)

    Parámetros

    poolSize

    Factores verticales y horizontales por los que reducir la escala.

    strides

    Los pasos.

    padding

    El acolchado.

  • Crea una capa de agrupación promedio con el tamaño y la zancada de la ventana de agrupación especificados. Todos los tamaños de agrupaciones y zancadas son iguales.

    Declaración

    public init(poolSize: Int, strides: Int, padding: Padding = .valid)