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