摘要：注：维度参数为1、dim为索引位置 一、扩展维度 data1 = data.unsqueeze(dim=n) 二、压缩维度 data2 = data.squeeze(dim=n) 阅读全文

摘要：一、numpy_array 转 torch_tensor import torch torch_data = torch.from_numpy(numpy_data) 二、torch_tensor 转 numpy_array 1 、 numpy_data = torch_data.numpy() 2 阅读全文

摘要：.pt, .pth, .pkl 仅仅是后缀不同 一、只保存模型权重参数，不保存模型结构 保存： torch.save(model.state_dict(), 'mymodel.pth') 加载： torch.load('mymodel.pth路径') #加载后是一个OrderedDict类型 调用： 阅读全文

摘要：一、利用scipy.io模块 二、io.loadmat方法读取 读取到的.mat文件是一个字典 可以通过matr.keys()查看keys 三、保存为.npy/.npz 1、.npy .npy文件 通过 np.load()方法加载 格式为array 2、.npz train_x = train_x 阅读全文