将带有字符串键和pytorch张量列表的字典保存为值

import pickle
import torch

# 创建包含字符串键和PyTorch张量列表的字典
data_dict = {
    "key1": [torch.tensor([1, 2, 3]), torch.tensor([4, 5, 6])],
    "key2": [torch.tensor([7, 8, 9]), torch.tensor([10, 11, 12])]
}

# 保存字典到文件
with open("data_dict.pkl", "wb") as file:
    pickle.dump(data_dict, file)

# 从文件加载字典
with open("data_dict.pkl", "rb") as file:
    loaded_dict = pickle.load(file)

# 打印加载后的字典
print(loaded_dict)