tf.keras.layers.Conv2DTranpose
backends для tf.nn.conv2d_transpose
через tf.keras.backend.conv2d_transpose
.
Для вычисления аргумента output_shape
для tf.nn.conv2d_transpose
используется функция deconv_output_length
(здесь определено ):
def deconv_output_length(input_length, filter_size, padding, stride):
"""Determines output length of a transposed convolution given input length.
Arguments:
input_length: integer.
filter_size: integer.
padding: one of "same", "valid", "full".
stride: integer.
Returns:
The output length (integer).
"""
if input_length is None:
return None
input_length *= stride
if padding == 'valid':
input_length += max(filter_size - stride, 0)
elif padding == 'full':
input_length -= (stride + filter_size - 2)
return input_length