本文简要介绍python语言中 torch.masked_select
的用法。
用法:
torch.masked_select(input, mask, *, out=None) → Tensor
返回一个新的一维张量,它根据布尔掩码
mask
索引input
张量,这是一个BoolTensor
。mask
张量和input
张量的形状不需要匹配,但它们必须是可广播的。注意
返回的张量不使用与原始张量相同的存储
例子:
>>> x = torch.randn(3, 4) >>> x tensor([[ 0.3552, -2.3825, -0.8297, 0.3477], [-1.2035, 1.2252, 0.5002, 0.6248], [ 0.1307, -2.0608, 0.1244, 2.0139]]) >>> mask = x.ge(0.5) >>> mask tensor([[False, False, False, False], [False, True, True, True], [False, False, False, True]]) >>> torch.masked_select(x, mask) tensor([ 1.2252, 0.5002, 0.6248, 2.0139])
相关用法
- Python PyTorch maskrcnn_resnet50_fpn用法及代码示例
- Python PyTorch matrix_rank用法及代码示例
- Python PyTorch max用法及代码示例
- Python PyTorch matrix_exp用法及代码示例
- Python PyTorch matmul用法及代码示例
- Python PyTorch matrix_power用法及代码示例
- Python PyTorch maximum用法及代码示例
- Python PyTorch matrix_norm用法及代码示例
- Python PyTorch make_tensor用法及代码示例
- Python PyTorch monitored_barrier用法及代码示例
- Python PyTorch mean用法及代码示例
- Python PyTorch multinomial用法及代码示例
- Python PyTorch meshgrid用法及代码示例
- Python PyTorch mm用法及代码示例
- Python PyTorch mv用法及代码示例
- Python PyTorch min用法及代码示例
- Python PyTorch msort用法及代码示例
- Python PyTorch mode用法及代码示例
- Python PyTorch movedim用法及代码示例
- Python PyTorch minimum用法及代码示例
- Python PyTorch multi_dot用法及代码示例
- Python PyTorch mul用法及代码示例
- Python PyTorch movielens_25m用法及代码示例
- Python PyTorch multigammaln用法及代码示例
- Python PyTorch movielens_20m用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.masked_select。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。