在mask中定义true，保留与其进行运算的tensor里的部分内容，相当于投影的功能。
mask与tensor的维度可以不相同的，但是对应的长度一定要相同，也就是要有一一对应的部分；
结果的维度 = tensor维度 - mask维度 + 1
以下是参考连接的例子，便于理解：
本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：tensorflow的boolean_mask函数 - Python技术站
微信扫一扫 
支付宝扫一扫
限制tensorflow训练时的显存设置，以及可见显卡号
上一篇
2023年4月6日
使用TensorFlow进行中文自然语言处理的情感分析
下一篇
2023年4月6日
相关文章
-
[PyTorch] torch.squeee 和 torch.unsqueeze()
torch.squeeze torch.squeeze(input, dim=None, out=None) → Tensor 分为两种情况: 不指定维度 或 指定维度 不指定维度 input: (A, B, 1, C, 1, D) output: (A, B, C, D) Example >>> x = torch.zeros(2, 1,…
-
PyTorch代码调试利器: 自动print每行代码的Tensor信息
本文介绍一个用于 PyTorch 代码的实用工具 TorchSnooper。作者是TorchSnooper的作者，也是PyTorch开发者之一。 GitHub 项目地址： https://github.com/zasdfgbnm/TorchSnooper 大家可能遇到这样子的困扰：比如说运行自己编写的 PyTorch 代码的时候，PyTorch 提示你说…
-
GAN 生成对抗模型详解
Goodfellow 等人提出来的 GAN 是通过对抗过程估计生成模型的新框架。在这种框架下，我们需要同时训练两个模型，即一个能捕获数据分布的生成模型 G 和一个能估计数据来源于真实样本概率的判别模型 D。生成器 G 的训练过程是最大化判别器犯错误的概率，即判别器误以为数据是真实样本而不是生成器生成的假样本。因此，这一框架就对应于两个参与者的极小极大博弈（m…
-
基于候选区域的深度学习目标检测算法R-CNN，Fast R-CNN，Faster R-CNN
参考文献 [1]Rich feature hierarchies for accurate object detection and semantic segmentation [2]Fast R-CNN [3]Faster R-CNN: towards real-time object detection with region proposal netw…
-
运用TensorFlow进行简单实现线性回归、梯度下降示例
下面是“运用TensorFlow进行简单实现线性回归、梯度下降”的完整攻略，包含两个实际示例说明： 实现线性回归 在使用 TensorFlow 实现线性回归时，通常分为以下几个步骤： 导入必要的库： import tensorflow as tf import numpy as np 准备数据，包括样本数据集 X 和标签数据集 Y。在这里，我们将使用随机生成…
-
Tensorflow环境安装记录–无法识别GPU的问题
1、镜像： -i http://pypi.douban.com/simple –trusted-host pypi.douban.com 2、版本信息（红色标注为我电脑的配置信息） 说明：在安装tensorflow-gpu环境时，一定要注意版本信息的对应，否则会出现各种奇葩的问题。 例如，我在安装tensorflow_gpu，由于默认安装的是最新的版本2.…
-
LSTM循环神经网络理解
1.循环神经网络 A是一组神经网络（可以理解为一个网络的自循环），它的工作是不停的接收并且输出。从图中可以看出A允许将信息不停的再内部循环，这样使得它可以保证每一步的计算都保存以前的信息
-
优化算法-从梯度下降到深度学习非凸优化
一、数学优化 1.1 定义 Mathematical Optimization（数学优化）问题，亦称最优化问题，是指在一定约束条件下，求解一个目标函数的最大值（或最小值）问题。 根据输入变量 ? 的值域是否为实数域，数学优化问题可以分为离散优化问题和连续优化问题． 在连续优化问题中，根据是否有变量的约束条件，可以将优化问题分为无约束优化问题和约束优化问题． …
