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

将带有字符串键和PyTorch张量列表的字典保存为值，可以使用Python中的pickle模块进行序列化和反序列化操作。pickle模块可以将Python对象转换为字节流，以便于存储和传输。

以下是一个示例代码，演示如何保存和加载包含字符串键和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)

在上述代码中，我们首先创建了一个包含字符串键和PyTorch张量列表的字典data_dict。然后，我们使用pickle.dump()函数将字典保存到名为"data_dict.pkl"的文件中。接下来，我们使用pickle.load()函数从文件中加载字典，并将其存储在loaded_dict变量中。最后，我们打印加载后的字典。