tf.pad()函数理解

3.tf.pad()

函数定义

tf.pad(
    tensor,
    paddings,
    mode='CONSTANT',
    name=None,
    constant_values=0
)
tensor      被填充的张量
paddings    填充的格式
mode        填充模式:"CONSTANT"、"REFLECT"、"SYMMETRIC"
name        该操作张量的名称
constant_values 用于在"CONSTANT"模式下,设置的填充值

padding它必须是 [N, 2] 形式,N代表张量的阶(也就是说tensor是几维的,那么N就是几), 2代表必须是2列,比如

tensor是2*3两维的,tensor = [[1,2,3],[4,5,6]]

那么padding必须要是2*2的,padding的列数一定是2.

padding = [[1,2],[1,1]] #第一行[1,2]表示在tensor的第一维上面操作,1代表在第一维最前面加一行全0;2代表在第一维的后面加2行全0.。

换句话说,padding一定是N行2列的,第j行,代表对tensor的第j维操作,两列的第一列是在j维的最前面加0的个数,第二列是在j维的最后面添0的个数。

猜你喜欢

转载自blog.csdn.net/ifreewolf_csdn/article/details/88934650
今日推荐