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})
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})