在mask中定义true，保留与其进行运算的tensor里的部分内容，相当于投影的功能。
mask与tensor的维度可以不相同的，但是对应的长度一定要相同，也就是要有一一对应的部分；
结果的维度 = tensor维度 - mask维度 + 1
以下是参考连接的例子，便于理解：
本站文章如无特殊说明，均为本站原创，如若转载，请注明出处：tensorflow的boolean_mask函数 - Python技术站
微信扫一扫 
支付宝扫一扫
限制tensorflow训练时的显存设置，以及可见显卡号
上一篇
2023年4月6日
使用TensorFlow进行中文自然语言处理的情感分析
下一篇
2023年4月6日
相关文章
-
数论入门1——莫比乌斯函数,欧拉函数,狄利克雷卷积,线性筛,莫比乌斯反演,杜教筛
一个菜鸡对数论的一点点理解… 莫比乌斯函数 定义函数\(\mu(n)\)为： 当n有平方因子时，\(\mu(n)=0\)。 当n没有平方因子时，\(\mu(n)=(-1)^{\omega(n)}\)，\(\omega(n)\)表示n不同质因子的个数。 性质1： \(\sum_{d|n}\mu(d)=[n=1]\) 证明：我们把n分解质因数，则原式\(=(…
-
PyTorch实现简单的变分自动编码器VAE
在上一篇博客中我们介绍并实现了自动编码器，本文将用PyTorch实现变分自动编码器(Variational AutoEncoder, VAE)。自动变分编码器原理与一般的自动编码器的区别在于需要在编码过程增加一点限制，迫使它生成的隐含向量能够粗略的遵循标准正态分布。这样一来，当需要生成一张新图片时，只需要给解码器一个标准正态分布的隐含随机向量就可…
-
deepdetect 用c++11写的机器学习caffe和XGBoost API 接口
https://github.com/beniz/deepdetect DeepDetect (http://www.deepdetect.com/) is a machine learning API and server written in C++11. It makes state of the art machine learning easy…
-
Tensorflow中tf.ConfigProto()的用法详解
在TensorFlow中，我们可以使用tf.ConfigProto()方法配置会话的参数，例如指定使用GPU进行计算、设置GPU的显存使用方式等。本文将详细讲解tf.ConfigProto()方法的用法，并提供两个示例说明。 示例1：指定使用GPU进行计算 以下是指定使用GPU进行计算的示例代码： import tensorflow as tf # 指定使用…
-
pytorch保存模型等相关参数，利用torch.save()，以及读取保存之后的文件
转载自： https://www.cnblogs.com/qinduanyinghua/p/9311410.html 假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch，我们要保存模型参数，优化器参数以及epoch 一、 1. 先建立一个…
-
张量（Tensor）、标量（scalar）、向量（vector）、矩阵（matrix）
张量（Tensor）：Tensor = multi-dimensional array of numbers 张量是一个多维数组，它是标量，向量，矩阵的高维扩展 ，是一个数据容器，张量是矩阵向任意维度的推广 注意，张量的维度（dimension）通常叫作轴（axis）, 张量轴的个数也叫作阶（rank）］ 标量（scalar）：只有一个数字的张量叫标量（也叫…
-
取出预训练模型中间层的输出(pytorch)
1 遍历子模块直接提取 对于简单的模型，可以采用直接遍历子模块的方法，取出相应name模块的输出，不对模型做任何改动。该方法的缺点在于，只能得到其子模块的输出，而对于使用nn.Sequensial()中包含很多层的模型，无法获得其指定层的输出。 示例 resnet18取出layer1的输出 from torchvision.models import res…
-
pytorch下的lib库 源码阅读笔记（1）
置顶：将pytorch clone到本地，查看initial commit，已经是麻雀虽小五脏俱全了，非常适合作为学习模板。 2017年12月7日01:24:15 2017-10-25 17:51 参考了知乎问题 如何有效地阅读PyTorch的源代码？ 相关回答 按照构建顺序来阅读代码是很聪明的方法。 1，TH中最核心的是THStorage、THTen…
