@frozen
public struct TransposedConv2D<Scalar> : Layer where Scalar : TensorFlowFloatingPoint
Uma camada de convolução transposta 2-D (por exemplo, convolução transposta espacial sobre imagens).
Essa camada cria um filtro de convolução que é transposto-convolvido com a entrada da camada para produzir um tensor de saídas.
O kernel de convolução 4-D.
Declaração
public var filter: Tensor<Scalar>
O vetor de polarização.
Declaração
public var bias: Tensor<Scalar>
A função de ativação por elemento.
Declaração
@noDerivative public let activation: Activation
Os passos da janela deslizante para dimensões espaciais.
Declaração
@noDerivative public let strides: (Int, Int)
O algoritmo de preenchimento para convolução.
Declaração
@noDerivative public let padding: Padding
A propriedade paddingIndex nos permite manipular a computação com base no preenchimento.
Declaração
@noDerivative public let paddingIndex: Int
Cria um
TransposedConv2D
camada com o filtro especificado, a polarização, a função de activação, passos, e estofamento.Declaração
public init( filter: Tensor<Scalar>, bias: Tensor<Scalar>? = nil, activation: @escaping Activation = identity, strides: (Int, Int) = (1, 1), padding: Padding = .valid )
Parâmetros
filter
Um tensor de 4-D da forma
[height, width, output channel count, input channel count]
.bias
O tensor de viés de forma
[output channel count]
.activation
A função de ativação por elemento.
strides
Os passos da janela deslizante para dimensões espaciais.
padding
O algoritmo de preenchimento para convolução.
Cria um
TransposedConv2D
camada com a forma especificada filtro, passos, preenchimento, e a função de activação elemento a elemento.Declaração
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
Um tensor de 4-D da forma
[width, height, input channel count, output channel count]
.strides
Os passos da janela deslizante para dimensões espaciais.
padding
O algoritmo de preenchimento para convolução.
activation
A função de ativação por elemento.
filterInitializer
Inicializador a ser usado para os parâmetros do filtro.
biasInitializer
Inicializador a ser usado para os parâmetros de polarização.