在mask中定义true，保留与其进行运算的tensor里的部分内容，相当于投影的功能。
mask与tensor的维度可以不相同的，但是对应的长度一定要相同，也就是要有一一对应的部分；
结果的维度 = tensor维度 - mask维度 + 1
以下是参考连接的例子，便于理解：
本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：tensorflow的boolean_mask函数 - Python技术站
微信扫一扫 
支付宝扫一扫
限制tensorflow训练时的显存设置，以及可见显卡号
上一篇
2023年4月6日
使用TensorFlow进行中文自然语言处理的情感分析
下一篇
2023年4月6日
相关文章
-
循环神经网络之LSTM
1、从LSTM的经典图入手 可以看到中间的 cell 里面有四个黄色小框:– 每一个小黄框代表一个前馈网络层，其实就是经典的神经网络的结构– 这个cell神经元个数和隐藏层个数皆可以设置– 其中1、2、4层的**函数是 sigmoid，第三层的**函数是 tanh。 2、对LSTM的理解 1）、cell 的状态是一个向量，是有多个值的 — cell在…
-
PyTorch项目使用TensorboardX进行训练可视化
什么是TensorboardX Tensorboard 是 TensorFlow 的一个附加工具，可以记录训练过程的数字、图像等内容，以方便研究人员观察神经网络训练过程。可是对于 PyTorch 等其他神经网络训练框架并没有功能像 Tensorboard 一样全面的类似工具，一些已有的工具功能有限或使用起来比较困难 (tensorboard_logger, …
-
机器学习（三）梯度下降与拟牛顿
这节课的推导真心hold不住了。按照自己的理解记下仅看明白的东西吧。或许还有第二遍、第三遍整理呢。 主要讲了两个问题： 学习率α如何确定？ 使用固定的学习率还是变化的学习率？ 学习率设置为多大比较好？ 下降的方向如何处理，除了梯度方向，有没有其他方向？ 可行方向和梯度方向有何关系？ 先上结论： 使用固定的学习率还是变化的学习率？ 使用变化的学习率…
-
目标检测（6）—IOU和非极大值抑制
转载自：https://www.cnblogs.com/zyly/p/9245451.html 网易云课堂—吴恩达深度学习目标检测课程内容 你如何判断对象检测算法运作良好呢？在这一节中，你将了解到并交比函数，可以用来评价对象检测算法。 一 并交比(Intersection over union ) 在对象检测任务中，你希望能够同时定位对象，所以如果实际边界框…
-
keras 序列模型
教程概述本教程分为5个部分; 他们是： TimeDistributed层序列学习问题用于序列预测的一对一LSTM用于序列预测的多对一LSTM(不含TimeDistributed)用于序列预测的多对多LSTM(带TimeDistributed)环境本教程假设你已经安装了带SciPy的Python 2或Python 3开发环境，以及NumPy和Pandas。 该…
-
TensorFlow？PyTorch？Paddle？AI工具库生态之争：ONNX将一统天下
AI诸多工具库工具库之间的切换，是一件耗时耗力的麻烦事。ONNX 即应运而生，使不同人工智能框架（如PyTorch、TensorRT、MXNet）可以采用相同格式存储模型数据并交互，极大方便了算法及模型在不同的框架之间的迁移，带来了AI生态的自由流通。… ? 作者：韩信子@ShowMeAI? 深度学习实战系列：https://www.showmeai.t…
-
Tensorflow使用GPU训练
确认显卡驱动正确安装： (notebook) [wuhf@aps ~]$ nvidia-smi Thu Aug 20 18:07:33 2020 +—————————————————————————–+ | NVIDIA-SMI 430.50 Driver …
-
计算机视觉—2.1—卷积与模板操作
理论推导见笔记本4—2015 5 21 由matlab生成一个高斯核 f = fspecial(‘gaussian’,[11 11],10); 可以生成的模板的种类。 其中[11,11]表示高斯核的尺寸，10是标准差，单位是“像素”。 单位是像素这个非常重要，“像素”的意义在于这个核跨越多少个像素后，它的作用衰竭90%。 这是跨越十个像素衰减90% 这是跨越…
