使用1而不是零的Tensorflow填充

Question

Tensorflow函数似乎只使用零填充（例如，tf.nn.conv2d函数）。

是否有一种简单的方法（需要对现有代码进行最小的更改）才能运行相同的函数，填充为1或任意其他任意数字而不是零？

Answer 1

最简单的方法是将conv2d中的填充从“SAME”更改为“VALID”并自行执行填充。

img = tf.ones([img_height+padding,img_width+padding, channels])
img[padding//2:img_height, padding//2:img_width] = old_img
result = conv2d(img, kernel, padding="VALID")