用 tf.pad() 填充 MNIST 数据集

padding MNIST dataset with tf.pad()

如何用 tf.pad() 填充大小为 (?,28,28,1) 的 MNIST 数据集图像并在 tensorflow 中使其成为 (?,32,32,1)

这要看你打算怎么垫了。例子是这样的:

# Assuming input is a tensor with shape (?, 28, 28, 1)
output = tf.pad(input, [[0, 0], [2,2], [2,2], [0,0]])
# print(tf.shape(output)) should be (?, 32, 32, 1)