tf.pad()简单使用

理解源自：https://blog.csdn.net/banana1006034246/article/details/76585355

函数格式:

tf.pad(input, paddings)

input--输入张量

paddings--padding的位置

 

1.当input为一维数据的时候：

paddings = [[2,5]]  : 对数据的第一维度在前面添加2项，后面添加5项

import tensorflow as tf
with tf.Session() as sess:
    input = [1,2,3]
    paddings = [[2,5]]
    print(sess.run(tf.pad(input,paddings)))

[0 0 1 2 3 0 0 0 0 0]

 

2.当input为一维数据的时候：

paddings = [[2,5], [1,3]]  : 对数据的第一维度在前面添加2项，后面添加5项。对数据的第二维度在前面添加1项，后面添加3项。

import tensorflow as tf
with tf.Session() as sess:
    input = [[1,2,3],[4,5,6]]
    paddings = [[2,5], [1,3]]
    print(sess.run(tf.pad(input,paddings)))

[[0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0]
 [0 1 2 3 0 0 0]
 [0 4 5 6 0 0 0]
 [0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0]]