Pytorch 保存tensor到.mat文件

Pytorch 保存tensor到.mat文件

import torch
import scipy.io as io
tensor = torch.ones(1, 2, 64)
tensor = tensor.cpu().detach().numpy()
#result = np.array(tensor)
io.savemat('save.mat', {'result':result})

Pytorch 保存tensor到.mat文件_第1张图片

你可能感兴趣的:(PyTorch学习,pytorch,python,numpy)