@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).
Esta 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 elemento a elemento.
Declaração
@noDerivative public let activation: Activation
Os avanços 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 lidar com a computação com base no preenchimento.
Declaração
@noDerivative public let paddingIndex: Int
Cria uma camada
TransposedConv2D
com o filtro, polarização, função de ativação, avanços e preenchimento especificados.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 4-D de forma
[height, width, output channel count, input channel count]
.bias
O tensor de polarização de forma
[output channel count]
.activation
A função de ativação elemento a elemento.
strides
Os avanços da janela deslizante para dimensões espaciais.
padding
O algoritmo de preenchimento para convolução.
Cria uma camada
TransposedConv2D
com a forma de filtro especificada, avanços, preenchimento e função de ativação por 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 4-D de forma
[width, height, input channel count, output channel count]
.strides
Os avanços da janela deslizante para dimensões espaciais.
padding
O algoritmo de preenchimento para convolução.
activation
A função de ativação elemento a elemento.
filterInitializer
Inicializador a ser usado para os parâmetros de filtro.
biasInitializer
Inicializador a ser usado para os parâmetros de polarização.