pytorch中mask操作之torch.masked

# Copyright 2018 CVTE . All Rights Reserved.
# coding: utf-8from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
import torch# torch.masked_select(input, mask, *, out=None) → Tensor
# 返回input中mask为True的元素，组成1维tensor
# input 和mask的维度不一定要相同,但是需要可以扩展成同样维度x = torch.randn(3, 4)
print(x)
"""
tensor([[-0.4152, -1.1952,  0.5442,  1.3843],[ 0.5764,  0.6684, -0.6330,  0.4646],[ 1.1162, -0.8730,  0.0285,  0.7387]])
"""mask = x.ge(0.5)
print(mask)"""
tensor([[0, 0, 1, 1],[1, 1, 0, 0],[1, 0, 0, 1]], dtype=torch.uint8)
"""print(torch.masked_select(x, mask))"""
tensor([0.5442, 1.3843, 0.5764, 0.6684, 1.1162, 0.7387])
"""# input 和mask的维度不一定要相同,但是需要可以扩展成同样维度
print(torch.masked_select(x, torch.tensor([False, False, True, False])))
"""
tensor([ 0.5442, -0.6330,  0.0285])
"""
print(torch.masked_select(x, torch.tensor([[False, False, True, False]])))
"""
tensor([ 0.5442, -0.6330,  0.0285])
"""
print(torch.masked_select(x, torch.tensor([[False], [False], [True]])))
"""
tensor([ 1.1162, -0.8730,  0.0285,  0.7387])
"""

pytorch中mask操作之torch.masked_select相关推荐

Pytorch中tensor维度和torch.max()函数中dim参数的理解
Pytorch中tensor维度和torch.max()函数中dim参数的理解维度参考了 https://blog.csdn.net/qq_41375609/article/details/106 ...
PyTorch中Variable变量与torch.autograd.Variable
一.了解Variable 顾名思义,Variable就是变量的意思.实质上也就是可以变化的量,区别于int变量,它是一种可以变化的变量,这正好就符合了反向传播,参数更新的属性. 具体来说,在pyt ...
pyTorch中tensor运算
文章目录 PyTorch的简介 PyTorch中主要的包 PyTorch的安装使用GPU的原因使数据在GPU上运行什么使Tensor(张量) 一些术语介绍 Tensor的属性介绍(Rank,ax ...
深入理解Pytorch负对数似然函数（torch.nn.NLLLoss)和交叉熵损失函数（torch.nn.CrossEntropyLoss)
在看Pytorch的交叉熵损失函数torch.nn.CrossEntropyLoss官方文档介绍中,给出的表达式如下.不免有点疑惑为何交叉熵损失的表达式是这个样子的 loss ⁡ ( y , clas ...
Pytorch中维度dim的理解使用
0 引言 pytorch中的维度dim主要被用在torch.softmax和torch.max等等函数中.理清dim的意思对于正确使用这些函数有重要意义. 1 相关博文: Pytorch笔记:维度di ...
pytorch中的MSELoss函数
基本概念均方误差(mean square error, MSE),是反应估计量与被估计量之间差异程度的一种度量,设ttt是根据子样确定的总体参数θ\thetaθ的一个估计量,(θ−t)2(\thet ...
opencv和pytorch中的warp操作函数：cv2.warpAffine, torch.nn.functional.grid_sample, cv2.warpPerspective
关于图像的warp操作是指利用一个旋转缩放矩阵对图像进行操作. 常见的操作有,平移,绕某个点旋转,缩放. opencv中有getRotationMatrix2D,warpAffine, getAffi ...
tensor flow lstm 图像一条直线_【开源计划】图像配准中变形操作（Warp）的pytorch实现...
前言按照开源计划的预告,这次我来分享图像配准流程中的变形操作的代码实现.首先我们先来回顾一下配准的流程,我们以这篇Unsupervised End-to-end Learning for Defor ...
pytorch 中利用自定义函数 get_mask_from_lengths(lengths, max_len)获取每个batch的mask
在pytorch中,经常会需要通过batch进行批量处理数据,由于每个batch中各个样本之间存在差异,经常会需要进行先padding后mask的操作. 尤其是在自然语言处理任务中,每个batch中的 ...

pytorch中mask操作之torch.masked_select

pytorch中mask操作之torch.masked_select相关推荐

最新文章

热门文章