Pytorch – masked_fill方法参数详解与使用
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Pytorch – masked_fill方法参数详解与使用
原文链接:https://www.stubbornhuang.com/2279/
发布于:2022年08月05日 9:10:14
修改于:2022年08月05日 9:10:14

1 torch.Tensor.masked_fill参数详解与使用
1.1 torch.Tensor.masked_fill参数详解
1. 函数形式
torch.Tensor.masked_fill(mask, value)
2. 函数功能
输入的mask需要与当前的基础Tensor的形状一致。
将mask中为True的元素对应的基础Tensor的元素设置为值value。
3. 函数参数
- mask:mask既可以是int型Tensor(值为0或者1)也可以是bool型Tensor(值为False或者True)
- value:float,填充的值
4. 函数返回值
返回填充后的Tensor
1.2 torch.Tensor.masked_fill使用示例
下面一个简单的例子说明masked_fill函数的使用,首先我们创建一个4x4的一个基础矩阵,然后创建一个4x4的对角矩阵,然后根据对角矩阵将对角线上的基础机矩阵的值全部设置为100,具体的代码如下所示。
import torch
if __name__ == '__main__':
tensor = torch.arange(0,16).view(4,4)
print('origin tensor:\n{}\n'.format(tensor))
mask = torch.eye(4,dtype=torch.bool)
print('mask tensor:\n{}\n'.format(mask))
tensor = tensor.masked_fill(mask,100)
print('filled tensor:\n{}'.format(tensor))
输出
origin tensor:
tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
mask tensor:
tensor([[ True, False, False, False],
[False, True, False, False],
[False, False, True, False],
[False, False, False, True]])
filled tensor:
tensor([[100, 1, 2, 3],
[ 4, 100, 6, 7],
[ 8, 9, 100, 11],
[ 12, 13, 14, 100]])
当前分类随机文章推荐
- Pytorch - 梯度累积/梯度累加trick,在显存有限的情况下使用更大batch_size训练模型 阅读239次,点赞0次
- Pytorch - torch.cat参数详解与使用 阅读1144次,点赞1次
- Pytorch - 没有使用with torch.no_grad()造成测试网络时显存爆炸的问题 阅读429次,点赞0次
- Pytorch - pad_sequence、pack_padded_sequence、pack_sequence、pad_packed_sequence参数详解与使用 阅读794次,点赞0次
- Pytorch - 模型保存与加载以及如何在已保存的模型的基础上继续训练模型 阅读426次,点赞0次
- Pytorch - 一文搞懂如何使用Pytorch构建与训练自定义深度学习网络(数据集自定义与加载,模型训练,模型测试,模型保存与加载) 阅读1015次,点赞2次
- Pytorch - torch.distributed.init_process_group函数详解 阅读463次,点赞0次
- Pytorch - 使用pytorch自带的Resnet作为网络的backbone 阅读203次,点赞0次
- Pytorch - torch.chunk参数详解与使用 阅读885次,点赞0次
- Pytorch - torch.nn.Module的parameters()和named_parameters() 阅读442次,点赞0次
全站随机文章推荐
- Python – 解决opencv-python使用cv2.imwrite()保存中文路径图片失败的问题 阅读1287次,点赞0次
- 资源分享 - Polygon Mesh Processing英文高清PDF下载 阅读6963次,点赞1次
- FFmpeg - PTS、DTS、时间基、时间戳详解 阅读312次,点赞0次
- C++ – UTF8编码下的全角字符转半角字符 阅读1565次,点赞0次
- Pytorch - torch.chunk参数详解与使用 阅读885次,点赞0次
- WordPress - 禁止非管理员登录后台 阅读1677次,点赞0次
- 书籍翻译 – Fundamentals of Computer Graphics, Fourth Edition,第6章 Transformation Matrices中文翻译 阅读1517次,点赞7次
- 工具网站推荐 - DLL‑FILES.COM帮你找到你的应用程序所缺失的dll文件 阅读2450次,点赞0次
- 资源分享 - GDI+教程(C++中文版 GDI+SDK中文参考手册)PDF下载 阅读2670次,点赞0次
- 资源分享 - Introduction to 3D Game Programming with DirectX 12 英文高清PDF下载 阅读2994次,点赞1次
评论
167