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 分钟前
零基础-动手学深度学习-6.6 卷积神经网络(LeNet)
人工智能·深度学习·cnn
无名工程师8 分钟前
AI 学习过程中各阶段的学习重点、时间规划以及不同方向的选择与建议等内容
人工智能·学习
FinAnalyzer9 分钟前
如何在 InsCodeAI 上搭建并使用 Jupyter Notebook 环境?
ide·python·jupyter
java1234_小锋11 分钟前
【NLP舆情分析】基于python微博舆情分析可视化系统(flask+pandas+echarts) 视频教程 - 微博文章数据可视化分析-文章分类下拉框实现
python·自然语言处理·flask
檀越剑指大厂11 分钟前
【Python系列】Flask 应用中的主动垃圾回收
开发语言·python·flask
檀越剑指大厂18 分钟前
【Python系列】使用 memory_profiler 诊断 Flask 应用内存问题
开发语言·python·flask
WXX_s30 分钟前
【OpenCV篇】OpenCV——03day.图像预处理(2)
人工智能·python·opencv·学习·计算机视觉
有才不一定有德1 小时前
深入剖析 MetaGPT 中的提示词工程:WriteCode 动作的提示词设计
人工智能·aigc·提示词工程
花月mmc1 小时前
CanMV-K230 AI学习笔记系列
人工智能·笔记·学习