标签：采样 索引 gather nd tensorflow2.0 tf print 部分 axis

import numpy as np
import tensorflow as tf

#   *************************   gather()根据索引提取数据    *****************************
a = tf.range(5)
print('原张量a:',a)
b = tf.gather(a,indices = [0,1,4])
print('根据索引提取出的数据b为:\n',b)
a = tf.reshape(tf.range(24),shape = (4,6))
print('多维张量a:',a)
b = tf.gather(a,axis = 0,indices = [0,1,3])
c = tf.gather(a,axis = 1,indices = [0,1,3])
print('axis = 0根据索引提取出的数据b为:\n',b)
print('axis = 1根据索引提取出的数据c为:\n',c)
print()
print('gather_nd()同时采样多个点')
print('原张量a:',a)
b = tf.gather_nd(a,[[0,1],[1,2],[2,0],[3,2]])
print('gather_nd采样后结果b:',b)

标签：采样,索引,gather,nd,tensorflow2.0,tf,print,部分,axis
来源： https://www.cnblogs.com/cxhzy/p/13386299.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。