pytorch保存数组到matlab .mat格式

网上找的代码都有一些问题，官网上的代码解决了遇到的错误:

from scipy.io import savemat
a = np.arange(20)
mdic = {"a": a, "label": "experiment"}
>> mdic
{'a': array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
    17, 18, 19]),
'label': 'experiment'}
savemat("matlab_matrix.mat", mdic)

来源: https://docs.scipy.org/doc/scipy/reference/generated/scipy.io.savemat.html

 

最终代码

result= np.array(result.detach().cpu())

result= np.squeeze(result)

mdic = {"a": result, "label": "experiment"}

scipy.io.savemat("matlab_matrix.mat", mdic)