【书籍阅读】Tensor学习笔记—索引

Tensor索引

tf.gather

数据结构『4,35,8』
4个班级,35个学生,8门课的成绩
从一个Tensor中一个维度出发抽取需要的特定的几个数据
【书籍阅读】Tensor学习笔记—索引_第1张图片

tf.gather_nd

从一个Tensor中多个维度出发抽取需要的特定的几个数据

【书籍阅读】Tensor学习笔记—索引_第2张图片其中a6和a7的区别在于一个是标量,一个是向量
【书籍阅读】Tensor学习笔记—索引_第3张图片选择多个数据后,需要将其拼接在一起(因为只有一个返回值)
另:多加一层括号就会多一维
【书籍阅读】Tensor学习笔记—索引_第4张图片

tf.boolean_mask

对于Tensor进行mask的过滤
【书籍阅读】Tensor学习笔记—索引_第5张图片mask为2x3的大小,所以是对前两维的数据进行过滤。

Ref

https://www.bilibili.com/video/BV1HV411q7xD?p=28&spm_id_from=pageDriver

你可能感兴趣的:(书籍阅读,深度学习,tensorflow)