已知一个字典中有多个类别key，每个类别value是一个torch.tensor(多个浮点型)，求这个字典所有value中的最大值
import torch
my_dict = {
'category1': torch.tensor([1.0, 2.0, 3.0]),
'category2': torch.tensor([4.0, 5.0, 6.0]),
'category3': torch.tensor([7.0, 8.0, 9.0])
}
# 使用列表推导和 torch.max() 找到最大值
max_value = max(torch.max(tensor).item() for tensor in my_dict.values())
print("最大值:", max_value)
标签：tensor,max,最大值,torch,value,字典 From： https://www.cnblogs.com/chentiao/p/17710714.html