Torsorflow相当于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))

Answer 2

有tf.keras.layers.UpSampling2D。我不确定，但我认为tf.image函数仅在CPU上实现。