Tensorflow中pad函数解析

来源:互联网 发布:多边形区域填充算法 编辑:程序博客网 时间:2024/06/01 08:54

函数原型:
tf.pad
pad(
tensor, # 输入数据
paddings,
mode=’CONSTANT’, # 可以有其它选项,但这里只说明填充0的情况
name=None
)

这里紧紧解释paddings的含义:
它是一个Nx2的列表,
在输入维度n上,则paddings[n, 0] 表示该维度内容前面加0的个数, 如对矩阵来说,就是行的最上面或列最左边加几排0
paddings[n, 1] 表示该维度内容后加0的个数。

举个例子:
输入 tensor 形如
[[ 1, 2],
[1, 2]]
而paddings = [[1, 1], [1, 1]]
则结果为
[[0, 0, 0, 0],
[0, 1, 2, 0],
[0, 1, 2, 0],
[0, 0, 0, 0]]

原创粉丝点击