Pytorch – masked_fill方法参数详解与使用
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]])
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Pytorch – masked_fill方法参数详解与使用
原文链接:https://www.stubbornhuang.com/2279/
发布于:2022年08月05日 9:10:14
修改于:2023年06月25日 20:45:33
当前分类随机文章推荐
- Pytorch - torch.topk参数详解与使用 阅读538次,点赞0次
- Pytorch - 使用pytorch自带的Resnet作为网络的backbone 阅读636次,点赞0次
- Pytorch - 梯度累积/梯度累加trick,在显存有限的情况下使用更大batch_size训练模型 阅读1031次,点赞0次
- Pytorch - 修改Pytoch中torchvision.models预置模型的方法 阅读379次,点赞0次
- Pytorch - 多GPU训练方式nn.DataParallel与nn.parallel.DistributedDataParallel的区别 阅读1478次,点赞0次
- Pytorch - torch.nn.Conv2d参数详解与使用 阅读1203次,点赞0次
- Pytorch - 使用Pyav解码视频文件并将视频帧转换为Pytorch tensor作为网络模型输入数据 阅读4525次,点赞0次
- Pytorch - transpose和permute函数的区别和用法 阅读1776次,点赞0次
- Pytorch - 模型微调时删除原有模型中的某一层的方法 阅读3223次,点赞0次
- Pytorch - 检测CUDA、cuDNN以及GPU版本的Pytorch是否安装成功、GPU显存测试 阅读4758次,点赞1次
全站随机文章推荐
- GoLand - Windows系统下载、安装、配置Go语言环境 阅读3211次,点赞0次
- 资源分享 - Handbook of Digital Image Synthesis - Scientific Foundations of Rendering 英文高清PDF下载 阅读1561次,点赞0次
- OpenCV - cv::VideoWriter::fourcc可支持的视频编码格式 阅读3885次,点赞0次
- Python3 - 正则表达式去除字符串中的特殊符号 阅读13911次,点赞1次
- C++11 - override关键字简要介绍 阅读2351次,点赞0次
- Python3爬虫 - requests的请求响应状态码(requests.status_code) 阅读9647次,点赞4次
- FFmpeg - ./configure编译参数全部总结和整理 阅读2625次,点赞3次
- Pytorch - torch.optim优化器 阅读1134次,点赞0次
- WordPress - 网站性能优化,延迟加载css和js文件 阅读149次,点赞0次
- 资源分享 - Jim Blinn's Corner - Notation, Notation, Notation 英文高清PDF下载 阅读2365次,点赞0次
评论
169