輸出tensor變量維度

print(a.shape)

輸出字典維度

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

輸出列表維度

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]