程序员经验分享-hwhale

tf.gather（）用法详解

2023-05-16

tf.gather(params, indices, validate_indices=None, axis=None, batch_dims=0, name=None )

 

请注意，在CPU上，如果找到超出范围的索引，则会返回错误。在GPU上，如果找到越界索引，则将0存储在相应的输出值中。

另请参阅tf.gather_nd。

 

import numpy as np
import tensorflow as tf

logits = [0,2,2,2,2,3,4,5,6,7,8,9]
target_id = [0,6]
selected_logits = tf.gather(logits, target_id)
with tf.Session() as sess:
    print(sess.run(selected_logits))

 

selected_logits取logits得第0位和第六位

 

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

tf.gather（）用法详解 的相关文章

随机推荐

  • 知识追踪在智能教育中的作用

    1 知识追踪的定义 现状及挑战 在上学的时候经常碰到过这样的状况 xff0c 老师会经常考试 xff0c 那么老师不能平白无故的给你考试 xff0c 为什么要考试呢 xff1f 他考试的主要目的就是为了掌握你对于某一知识的衡量水平 xff0

  • 吴恩达《神经网络与深度学习》课程笔记

    课堂笔记方便自己回忆 课程相关 课程地址见 https mooc study 163 com course 2001281002 info 该专项课程的Coursera地址 xff1a https www coursera org spec

  • matlab网页运行

    matlib软件巨大 xff0c 没有安装 找了几个在线网址 http www compileonline com execute matlab online php https octave online net

  • STM32入门教程——点亮一个LED

    STM32系列MCU被广泛的应用 xff0c 新学者想学习单片机或嵌入式时 xff0c 可以先从STM32入手 xff0c 本文介绍如何使用STM32单片机点亮一个LED 开发环境搭建 1 xff09 keil mdk安装 stm32系列M

  • MATLAB实验一 数字图像处理+函数调用笔记

    函数调用 https www cnblogs com gshang p 12763028 html 1 Write and test a MATLAB script file called lab1 plot m to plot the f

  • MATLAB学习之矩阵与幻方矩阵

    原文见https ww2 mathworks cn help matlab learn matlab matrices and magic squares html 上面为官方MATLAB原网址 xff0c 写此博客为方便复习回顾 矩阵和幻

  • matlab入门学习系列之表达式 变量运算符

    目录 变量 矩阵运算符 数组运算符 函数 表达式示例 变量 与大多数其他编程语言一样 xff0c MATLAB 语言提供数学表达式 xff0c 但与大多数编程语言不同的是 xff0c 这些表达式涉及整个矩阵 MATLAB 不需要任何类型声明

  • MATLAB入门学习系列之输入输出和索引

    取消输出 如果您在仅键入语句后按 Return 或 Enter xff0c MATLAB 会在屏幕上自动显示结果 但是 xff0c 如果使用分号结束行 xff0c MATLAB 会执行计算 xff0c 但不会显示任何输出 当生成大型矩阵时

  • MATLAB入门学习系列之基本绘图函数

    目录 创建绘图 在一幅图形中绘制多个数据集 指定线型和颜色 绘制线条和标记 将绘图添加到现有图形中 图窗窗口 在一幅图窗中显示多个绘图 控制轴 保存图窗 保存工作区数据 创建绘图 plot 函数具有不同的形式 xff0c 具体取决于输入参数

  • MATLAB入门学习系列之图像

    显示图像 图像数据 您可以将二维数值数组显示为图像 在图像中 xff0c 数组元素决定了图像的亮度或颜色 例如 xff0c 加载一个图像数组及其颜色图 xff1a gt gt load durer gt gt whos Name Size

  • 知识跟踪的深度知识跟踪和动态学生分类 Deep Knowledge Tracing and Dynamic Student Classification for Knowledge Tracing

    Deep Knowledge Tracing and Dynamic Student Classification for Knowledge Tracing xff08 译 xff09 知识跟踪的深度知识跟踪和动态学生分类 摘要 在智能辅

  • 知识追踪常见建模方法之IRT项目反应理论

    目录 A 项目反应理论 xff08 IRT item response theory xff09 概述 历史发展 特点 模型 A 项目反应理论 xff08 IRT item response theory xff09 概述 IRT理论即项目

  • MATLAB图像处理基本操作(1)

    matlib软件巨大 xff0c 没有安装 找了几个在线网址 http www compileonline com execute matlab online php https octave online net 从文件读取图像 a xf

  • Python学习系列之类的定义、构造函数 def __init__

    python def init self name等多参数 def init self 常见的两种类的定义方式如下 第一种 class Student def init self 两者之间的区别 self name 61 None self

  • ChatGPT，爆了！

    这段时间真是太刺激了 xff0c AI领域几乎每天都会爆出一个超震撼的产品 xff0c 有一种科幻马上要成现实的感觉 不知道大家朋友圈是什么样 xff0c 在整个创业的圈子里面 xff0c 几乎全是 AI 相关 就连 N 多年 xff0c

  • 分类回归模型评估常见方法及ROC AUC

    目录 模型评估常见方法 ROC和AUC定义 sklearn计算ROC具体实现 计算ROC需要知道的关键概念 1 分析数据 2 针对score xff0c 将数据排序 3 将截断点依次取为score值 3 1 截断点为0 1 sklearn

  • Coursera 吴恩达《Machine Learning》课堂笔记 + 作业

    记录一下最近学习的资源 xff0c 方便寻找 xff1a Github 上已经有人把作业整理成为 Python 的形式了 有 py 和 ipynb 两种格式 https github com nsoojin coursera ml py h

  • tensflow学习小知识tf.train.exponential_decay

    tf train exponential decay是tensflow1 X版本的2 版本使用以下语句 tf compat v1 train exponential decay 将指数衰减应用于学习率 tf compat v1 train

  • PyTorch学习系列之PyTorch：nn和PyTorch：optim优化

    PyTorch xff1a nn 在构建神经网络时 xff0c 我们经常考虑将计算分为几层 xff0c 其中一些层具有可学习的参数 xff0c 这些参数将在学习过程中进行优化 在TensorFlow xff0c 像包 Keras xff0c

  • tf.gather（）用法详解

    tf gather params indices validate indices 61 None axis 61 None batch dims 61 0 name 61 None 请注意 xff0c 在CPU上 xff0c 如果找到超出

热门标签