2023年9月18日
求一个字典的所有value中的最大值
摘要： 已知一个字典中有多个类别key，每个类别value是一个torch.tensor(多个浮点型)，求这个字典所有value中的最大值 import torch my_dict = { 'category1': torch.tensor([1.0, 2.0, 3.0]), 'category2': to 阅读全文
posted @ 2023-09-18 09:03 海_纳百川 阅读(16) 评论(0) 推荐(0) 编辑
本站总访问量次