1 torch.Tensor.masked_fill参数详解与使用

1.1 torch.Tensor.masked_fill参数详解

1. 函数形式

torch.Tensor.masked_fill(mask, value)

2. 函数功能
输入的maskmaskmask需要与当前的基础Tensor的形状一致。
将maskmaskmask中为True的元素对应的基础Tensor的元素设置为值valuevaluevalue。

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 torchif __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 - masked_fill方法参数详解与使用相关推荐

[pytorch]yolov3.cfg参数详解（每层输出及route、yolo、shortcut层详解）
文章目录 Backbone(Darknet53) 第一次下采样(to 208) 第二次下采样(to 104) 第三次下采样(to 52) 第四次下采样(to 26) 第五次下采样(to 13) YOL ...
pytorch nn.LSTM()参数详解
输入数据格式: input(seq_len, batch, input_size) h0(num_layers * num_directions, batch, hidden_size) c0(num ...
intent.putExtra()方法参数详解
[putExtra("A",B)中,AB为键值对,第一个参数为键名,第二个参数为键对应的值.顺便提一下,如果想取出Intent对象中的这些值,需要在你的另一个Activity中用g ...
JQuery中$.ajax()方法参数详解（转）
url: 要求为String类型的参数,(默认为当前页地址)发送请求的地址. type: 要求为String类型的参数,请求方式(post或get)默认为get.注意其他http请求方法,例如put和 ...
动态代理模式newProxyInstance及invoke方法参数详解
动态代理概述: Java提供的动态代理类Proxy: Proxy provides static methods for creating dynamic proxy classes and inst ...
Pytorch LSTM模型参数详解
本文主要依据 Pytorch 中LSTM官方文档,对其中的模型参数.输入.输出进行详细解释. 目录基本原理模型参数 Parameters 输入Inputs: input, (h_0, c_0) 输 ...
Assert.assertEquals()方法参数详解
junit.framework包下的Assert提供了多个断言方法. 主用于比较测试传递进去的两个参数. Assert.assertEquals();及其重载方法: 1. 如果两者一致, 程序继续往下 ...
java recordset_adodb.recordset.open(rs.open)方法参数详解
ADO Recordset 对象 ADO Recordset 对象用于容纳一个来自数据库表的记录集.一个 Recordset 对象由记录和列(字段)组成. 在 ADO 中,此对象是最重要且最常用于对数 ...
canvas的drawImage方法参数详解
引言: canvas很有意思的一项特性就是图像操作能力.可以用于动态的图像合成或者作为图形的背景,以及游戏界面(Sprites)等等.浏览器支持的任意格式的外部图片都可以使用,比如PNG.GIF或者J ...
html5 drawimage参数,小程序中canvas的drawImage方法参数详解
最近在开发小程序,海报生成的过程中,要在carvas中不断去添加图片,对小程序的drawImage参数不是很明确,这次解惑. 示例代码有三个版本的写法: drawImage(imageResourc ...

Pytorch - masked_fill方法参数详解与使用

1 torch.Tensor.masked_fill参数详解与使用

1.1 torch.Tensor.masked_fill参数详解

1.2 torch.Tensor.masked_fill使用示例

Pytorch - masked_fill方法参数详解与使用相关推荐

最新文章

热门文章