TensorFlow中与Keras函数UpSampling2D等效的函数是什么？

Question

我想使用Keras层：

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

我应该如何使用TensorFlow原生功能复制这种行为呢？我找不到相应的函数/层。

- Jonathan DEKHTIAR

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))

- amirbar

Answer 2

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

- nio1814