将tensor进行整理，参数为conf_thres，用python代码怎么写

假设要整理的tensor为`tensor_data`，参数`conf_thres`表示阈值，以下是用Python代码实现的示例：

import torch

def filter_tensor_by_confidence(tensor_data, conf_thres):
    # 获取置信度大于阈值的元素的索引
    idx = torch.where(tensor_data[:, :, 4] >= conf_thres)
    # 根据索引获取符合条件的元素
    filtered_tensor = tensor_data[idx]
    return filtered_tensor

在这个示例中，我们通过`torch.where`函数获取了置信度大于等于阈值的元素的索引，然后通过索引获取了符合条件的元素。这个函数返回一个新的tensor，其中包含符合条件的元素。