tf.pad学习

https://blog.csdn.net/yy_diego/article/details/81563160

https://www.tensorflow.org/api_docs/python/tf/pad

1.二维张量

pad(
    tensor,
    paddings,
    mode='CONSTANT',#constant默认就是0
    name=None
)

三行四列的1值，[1,2]表示对第一维，也就是行，第一行上面添加一行0，下面添加两行0，[3,4]对第二个维度，左边添加3列0，右边添加4列0.

# 创建一个二维变量,默认执行CONSTANT填充
vct = tf.Variable(tf.ones([3, 4]), name="vct")
# 指定填充方式,
pad1 = np.array([[1, 2], [3, 4]])
# tf.pad进行填充
vct_pad1 = tf.pad(vct, pad1, name='pad_1')
# 创建会话
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(vct))
    print(sess.run(vct_pad1))

 

 2.三维张量

//这个就更好理解了。

tsr = tf.Variable(tf.ones([2, 3, 4]), name="tsr")
pad2 = np.array([[1, 1], [0, 0], [0, 0]])
tsr_pad2 = tf.pad(tsr, pad2, name='pad_2')

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(tsr))
    print(sess.run(tsr_pad2))

[0,0]针对这个2维度，上下增加一个0，