Транспонированная свертка в tf.keras.layers - PullRequest
0 голосов
/ 16 марта 2019

Как я могу преобразовать эту функцию в эквивалентную реализацию tf.keras (Tensorflow 2.0)?

def deconv2d(inputs,out_channels, name=None):
batch, in_height, in_width, in_channels = [int(d) for d in inputs.get_shape()]
filter = tf.get_variable(name + "filter", [4, 4, out_channels, in_channels], initializer= tf.variance_scaling_initializer())
return tf.nn.conv2d_transpose(inputs, filter, [batch, in_height * 2, in_width * 2, out_channels], strides=[1,2,2,1],
                             padding='SAME', name= name)

1 Ответ

0 голосов
/ 16 марта 2019

TensoFlow 2.0 имеет tf.keras.layers.Conv2DTranspose. Keras - это API-интерфейс высокого уровня по умолчанию для TF 2.0, но он все еще имеет tf.nn.conv2d_transpose для приложений низкого уровня

https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/keras/layers/Conv2DTranspose

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...