使用tf.pad（）填充MNIST数据集

Question

如何使用(?,28,28,1)填充大小为tf.pad()的MNIST数据集图像，并在张量流中将其设为(?,32,32,1)？

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