PyTorch torch.masked_select 函数

PyTorch torch.masked_select 函数 Pytorch torch 参考手册 torch.masked_select 是 PyTorch 中用于根据布尔掩码选择元素的函数。函数定义 torch.masked_select(input, mask) 使用示例实例 [mycode4 type='python'] import torch x = torch.tensor([..

更新于 2026/7/16

教程详情页默认 noindex。

PyTorch torch.masked_select 函数

Pytorch torch 参考手册

torch.masked_select 是 PyTorch 中用于根据布尔掩码选择元素的函数。

函数定义

torch.masked_select(input, mask)

使用示例

实例

import torch

x = torch.tensor([1, 2, 3, 4, 5])
mask = torch.tensor([True, False, True, False, True])

# 根据掩码选择元素
result = torch.masked_select(x, mask)

print(result)

输出结果为：

tensor([1, 3, 5])

Pytorch torch 参考手册