Tensorflow 2.0 API Interface 个人理解
tf.gather() #从一个维度上,根据索引获取tensor数据的一部分 例如:tf.gather(a, axis = 0, indices = [2, 3]) 假设 a.shape ---------> [2, 35, 8] 上述操作,就是在a的0维度上获取,索引为2、3的数据 则tf.gather(a, axis = 0, indices ...
原创
2021-11-02 09:49:42 ·
612 阅读 ·
0 评论