tensorflow équivalent de la fonction Keras: UpSampling2D

Question

Je voudrais utiliser la couche Keras:

from keras.layers.convolutional import UpSampling2D 
x = UpSampling2D((2, 2))(x) 

Comment puis-je reproduire ce comportement avec tensorflow natif?

Je n'arrive pas à trouver une fonction/couche équivalente.

Answer 1

En supposant x est de forme (BATCH_SIZE, H, W, C), vous pouvez utiliser tf.image.resize_nearest_neighbor, qui est la mise en œuvre de back-end utilisé par keras:

x = tf.image.resize_nearest_neighbor(x, (2*H,2*W))