如何在TensorFlow中沿特定轴实现填充？

Question

我将拥有一个具有（16，256，256，3）形状的张量A（16是批处理大小，256是图像块的大小，3是图像块的通道数）。我想在TensorFlow中使用“ PadLayer”沿张量A的第二维和第三维（即256）填充零。

例如，填充结果应为（16，300，300，3）。我想问一下如何实现？如何在TensorFlow中设置“填充”参数。

非常感谢您的帮助！