tf.reduce_sum() and tf.where()的用法
2024-09-06 09:52:07
import tensorflow as tf
import numpy as np
sess=tf.Session()
a=np.ones((5,6))
c=tf.cast(tf.reduce_sum(a, axis=1),tf.bool) # tf.reduce_sum 表示按照axis方向求和
c=sess.run(c)
print('c=',c)
print('c.shape',c.shape)
k=tf.where(c) # 因为返回的是索引值为列表中
k=sess.run(k)
print('k=',k)
print('k.shape=',k.shape)
k_compare=tf.where(c)[:, 0]
k_compare=sess.run(k_compare)
print('k_compare=',k_compare)
print('k_compare.shape=',k_compare.shape) 结果如下:
最新文章
- web加密的基本概念
- Winform自定义控件基础(二)
- 【干货理解】理解javascript中实现MVC的原理
- connect VisualVM to Tomcat
- 如何使Session永不过期
- 20145208 《Java程序设计》第5周学习总结
- 跟开涛老师学shiro -- shiro简介
- UltraEdit20 注册
- apache 配置用户级目录
- C语言的学习-基础知识点
- POJ1094 拓扑排序
- 爬虫关于ip管理池的应用
- 【js】项目中有关时间的问题
- leaflet动态路径
- input 在 2 和 3 上的区别
- 第3章 Data语意学
- Excel VBA 连接各种数据库(二) VBA连接Oracle数据库
- 2/18 (pycharm 快捷键、循环、join语句)
- QC内部分享ppt
- ArrayList与List性能测试