select函数_PyTorch入门笔记masked_select选择函数

关注" AI机器学习与深度学习算法"公众号 选择" 星标 "公众号，原创干货，第一时间送达

masked_select

torch.masked_select(input,mask,out=None) 函数返回一个根据布尔掩码 (boolean mask) 索引输入张量的 1D 张量，其中布尔掩码和输入张量就是 torch.masked_select(input, mask, out = None) 函数的两个关键参数，函数的参数有：

• input(Tensor) - 需要进行索引操作的输入张量；
• mask(BoolTensor) - 要进行索引的布尔掩码
• out(Tensor, optional) - 指定输出的张量。比如执行 torch.zeros([2, 2], out = tensor_a)，相当于执行 tensor_a = torch.zeros([2, 2])；

「masked_sele