您的位置：首页 > 编程技术 > python  >  tensorflow中tf.slice和tf.gather切片函数的使用

tensorflow中tf.slice和tf.gather切片函数的使用

 更新时间：2020年4月22日 23:19  点击：1456

tf.slice(input_, begin, size, name=None)：按照指定的下标范围抽取连续区域的子集

tf.gather(params, indices, validate_indices=None, name=None)：按照指定的下标集合从axis=0中抽取子集，适合抽取不连续区域的子集

输出：

input = [[[1, 1, 1], [2, 2, 2]],
   [[3, 3, 3], [4, 4, 4]],
   [[5, 5, 5], [6, 6, 6]]]
tf.slice(input, [1, 0, 0], [1, 1, 3]) ==> [[[3, 3, 3]]]
tf.slice(input, [1, 0, 0], [1, 2, 3]) ==> [[[3, 3, 3],
           [4, 4, 4]]]
tf.slice(input, [1, 0, 0], [2, 1, 3]) ==> [[[3, 3, 3]],
           [[5, 5, 5]]]
           
tf.gather(input, [0, 2]) ==> [[[1, 1, 1], [2, 2, 2]],
        [[5, 5, 5], [6, 6, 6]]]

假设我们要从input中抽取[[[3, 3, 3]]]，这个输出在inputaxis=0的下标是1，axis=1的下标是0，axis=2的下标是0-2，所以begin=[1,0,0]，size=[1,1,3]。

假设我们要从input中抽取[[[3, 3, 3], [4, 4, 4]]]，这个输出在inputaxis=0的下标是1，axis=1的下标是0-1，axis=2的下标是0-2，所以begin=[1,0,0]，size=[1,2,3]。

假设我们要从input中抽取[[[3, 3, 3], [5, 5, 5]]]，这个输出在inputaxis=0的下标是1-2，axis=1的下标是0，axis=2的下标是0-2，所以begin=[1,0,0]，size=[2,1,3]。

假设我们要从input中抽取[[[1, 1, 1], [2, 2, 2]],[[5, 5, 5], [6, 6, 6]]]，这个输出在input的axis=0的下标是[0, 2]，不连续，可以用tf.gather抽取。input[0]和input[2]

以上这篇tensorflow中tf.slice和tf.gather切片函数的使用就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持猪先飞。

[!--infotagslink--]

相关文章

  • 解决tensorflow训练时内存持续增加并占满的问题

    今天小编就为大家分享一篇解决tensorflow训练时内存持续增加并占满的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22

  • win10安装tensorflow-gpu1.8.0详细完整步骤

    这篇文章主要介绍了win10安装tensorflow-gpu1.8.0详细完整步骤，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下...2020-04-22

  • macOS M1(AppleSilicon) 安装TensorFlow环境

    苹果为M1芯片的Mac提供了TensorFlow的支持，本文主要介绍了如何给使用M1芯片的macOS安装TensorFlow的环境，感兴趣的可以了解一下...2021-08-13

  • 详解c# 切片语法糖

    这篇文章主要介绍了c# 切片语法糖的相关资料，帮助大家更好的理解和学习c#，感兴趣的朋友可以了解下...2020-12-08

  • windows系统Tensorflow2.x简单安装记录(图文)

    这篇文章主要介绍了windows系统Tensorflow2.x简单安装记录(图文)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧...2021-01-18

  • TensorFlow神经网络学习之张量与变量概念

    这篇文章主要为大家介绍了TensorFlow神经网络学习的基本知识张量与变量概念详解，有需要的朋友可以借鉴参考下，希望能够有所帮助...2021-10-17

  • Tensorflow读取并输出已保存模型的权重数值方式

    今天小编就为大家分享一篇Tensorflow读取并输出已保存模型的权重数值方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看不看...2020-04-30

  • 完美解决TensorFlow和Keras大数据量内存溢出的问题

    这篇文章主要介绍了完美解决TensorFlow和Keras大数据量内存溢出的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-07-04

  • 在tensorflow下利用plt画论文中loss,acc等曲线图实例

    这篇文章主要介绍了在tensorflow下利用plt画论文中loss,acc等曲线图实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-06-16

  • Python通过TensorFLow进行线性模型训练原理与实现方法详解

    这篇文章主要介绍了Python通过TensorFLow进行线性模型训练原理与实现方法,结合实例形式详细分析了Python通过TensorFLow进行线性模型训练相关概念、算法设计与训练操作技巧,需要的朋友可以参考下...2020-04-27

  • 详解tf.device()指定tensorflow运行的GPU或CPU设备实现

    这篇文章主要介绍了详解tf.device()指定tensorflow运行的GPU或CPU设备实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧...2021-02-20

  • tensorflow实现对张量数据的切片操作方式

    今天小编就为大家分享一篇tensorflow实现对张量数据的切片操作方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22

  • 基于tensorflow for循环 while循环案例

    这篇文章主要介绍了基于tensorflow for循环 while循环案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-06-30

  • 从训练好的tensorflow模型中打印训练变量实例

    今天小编就为大家分享一篇从训练好的tensorflow模型中打印训练变量实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22

  • tensorflow -gpu安装方法（不用自己装cuda，cdnn）

    这篇文章主要介绍了tensorflow -gpu安装，史上最新最简单的途径（不用自己装cuda，cdnn）,非常不错，具有一定的参考借鉴价值，需要的朋友可以参考下...2020-04-22

  • tensorflow estimator 使用hook实现finetune方式

    今天小编就为大家分享一篇tensorflow estimator 使用hook实现finetune方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22

  • Go遍历struct,map,slice的实现

    本文主要介绍了Go语言遍历结构体、切片和字典的方法，对大家的学习具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧...2021-06-13

  • 浅谈tensorflow中张量的提取值和赋值

    今天小编就为大家分享一篇浅谈tensorflow中张量的提取值和赋值，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22

  • 使用TensorFlow-Slim进行图像分类的实现

    这篇文章主要介绍了使用TensorFlow-Slim进行图像分类的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧...2020-05-06

  • tensorflow查看ckpt各节点名称实例

    今天小编就为大家分享一篇tensorflow查看ckpt各节点名称实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧...2020-04-22