Tensorflow相當於Keras功能：UpSampling2D

Question

我想用Keras層：

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

我怎麼能複製與本地tensorflow這種行爲？

我無法找到等價的函數/圖層。

Answer 1

假設x是形狀(BATCH_SIZE, H, W, C)的，則可以使用tf.image.resize_nearest_neighbor，這是通過使用keras後端實現：

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