pytorch查询字典、列表维度

输出tensor变量维度

javascript 复制代码
print(a.shape)

输出字典维度

javascript 复制代码
for key, value in output_dict.items():
    if isinstance(value, torch.Tensor):
        print(f"{key} shape:", value.shape)

输出列表维度

javascript 复制代码
def get_list_dimensions(lst):
    # 基线条件:如果lst不是列表,返回空列表
    if not isinstance(lst, list):
        return []
    # 基线条件:如果lst是空列表,返回[0]
    if not lst:
        return [0]
    # 递归步骤:检查子列表的维度
    sub_dimensions = [get_list_dimensions(item) for item in lst if isinstance(item, list)]
    # 如果没有子列表,则当前列表是一维的
    if not sub_dimensions:
        return [len(lst)]
    # 确定所有子列表的最大维度
    max_dim = max(sub_dimensions, key=len)
    # 返回当前列表长度和子列表的最大维度
    return [len(lst)] + max_dim

# 测试函数
my_list = [[1, 2], [3, 4]]
print(get_list_dimensions(my_list))  # 应该输出 [2, 2]
相关推荐
子燕若水3 分钟前
Unreal Engine 5中的AI知识
人工智能
极限实验室1 小时前
Coco AI 实战(一):Coco Server Linux 平台部署
人工智能
杨过过儿1 小时前
【学习笔记】4.1 什么是 LLM
人工智能
巴伦是只猫1 小时前
【机器学习笔记Ⅰ】13 正则化代价函数
人工智能·笔记·机器学习
大千AI助手2 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配
AI生存日记2 小时前
百度文心大模型 4.5 系列全面开源 英特尔同步支持端侧部署
人工智能·百度·开源·open ai大模型
LCG元2 小时前
自动驾驶感知模块的多模态数据融合:时序同步与空间对齐的框架解析
人工智能·机器学习·自动驾驶
why技术2 小时前
Stack Overflow,轰然倒下!
前端·人工智能·后端
烛阴3 小时前
简单入门Python装饰器
前端·python