将下列numpy代码转化为纯pytorch代码： inds = np.where(scores > args.confidence_threshold)[:] boxes = boxes[inds] landms = landms[inds] scores = scores[inds]

请使用 PyTorch 的 mask 和 index_select 函数来替换 numpy 中的 where 和索引选择。

import torch

# 将 scores 转化为 PyTorch tensor
scores = torch.tensor(scores)

# 创建一个 mask，用来筛选出 scores 大于 args.confidence_threshold 的元素
mask = (scores > args.confidence_threshold)

# 使用 mask 和 index_select 函数来筛选出 boxes、landms 和 scores
boxes = boxes[mask]
landms = landms[mask]
scores = scores[mask]