如何检查张量值是否在不同的张量火炬中？

import torch

def check_tensor_in_tensor_bundles(target_value, tensor_bundles):
    for idx, tensor_bundle in enumerate(tensor_bundles):
        if torch.any(tensor_bundle == target_value):
            print(f"Target value {target_value} found in tensor bundle {idx}!")
            # 可以在这里执行其他逻辑操作
        else:
            print(f"Target value {target_value} not found in tensor bundle {idx}.")

# 示例使用
# 假设有3个张量火炬，每个张量火炬都是torch.Tensor类型
tensor_bundle_1 = torch.tensor([1, 2, 3])
tensor_bundle_2 = torch.tensor([4, 5, 6])
tensor_bundle_3 = torch.tensor([7, 8, 9])

# 将3个张量火炬放入一个列表中
tensor_bundles = [tensor_bundle_1, tensor_bundle_2, tensor_bundle_3]

# 检查值为5的张量是否在不同的张量火炬中
check_tensor_in_tensor_bundles(5, tensor_bundles)