TransposedConv2D

@frozen
public struct TransposedConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint

Una capa de convolución transpuesta 2-D (por ejemplo, convolución transpuesta espacial sobre imágenes).

Esta capa crea un filtro de convolución que se transpone-convoluciona con la entrada de la capa para producir un tensor de salidas.

  • El núcleo de convolución 4-D.

    Declaración

    public var filter: Tensor<Scalar>
  • El vector de sesgo.

    Declaración

    public var bias: Tensor<Scalar>
  • La función de activación por elementos.

    Declaración

    @noDerivative
    public let activation: Activation
  • Los pasos de la ventana deslizante para dimensiones espaciales.

    Declaración

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

    Declaración

    @noDerivative
    public let padding: Padding
  • La propiedad paddingIndex nos permite manejar el cálculo basado en el relleno.

    Declaración

    @noDerivative
    public let paddingIndex: Int
  • El tipo de función de activación por elementos.

    Declaración

    public typealias Activation = @differentiable (Tensor<Scalar>) -> Tensor<Scalar>
  • Crea un TransposedConv2D capa con el filtro especificado, el sesgo, la función de activación, zancadas, y el relleno.

    Declaración

    public init(
      filter: Tensor<Scalar>,
      bias: Tensor<Scalar>? = nil,
      activation: @escaping Activation = identity,
      strides: (Int, Int) = (1, 1),
      padding: Padding = .valid
    )

    Parámetros

    filter

    A 4-D tensor de la forma [height, width, output channel count, input channel count] .

    bias

    El tensor de polarización de la forma [output channel count] .

    activation

    La función de activación por elementos.

    strides

    Los pasos de la ventana deslizante para dimensiones espaciales.

    padding

    El algoritmo de relleno para convolución.

  • Devuelve la salida obtenida 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 devuelto

    La salida.

  • Crea un TransposedConv2D capa con la forma especificada filtro, zancadas, relleno, y la función de activación de elemento a elemento.

    Declaración

    public init(
      filterShape: (Int, Int, Int, Int),
      strides: (Int, Int) = (1, 1),
      padding: Padding = .valid,
      activation: @escaping Activation = identity,
      useBias: Bool = true,
      filterInitializer: ParameterInitializer<Scalar> = glorotUniform(),
      biasInitializer: ParameterInitializer<Scalar> = zeros()
    )

    Parámetros

    filterShape

    A 4-D tensor de la forma [width, height, input channel count, output channel count] .

    strides

    Los pasos de la ventana deslizante para dimensiones espaciales.

    padding

    El algoritmo de relleno para convolución.

    activation

    La función de activación por elementos.

    filterInitializer

    Inicializador que se utilizará para los parámetros del filtro.

    biasInitializer

    Inicializador que se utilizará para los parámetros de polarización.