import torch import numpy as np np_data=np.arange(6).reshape(2,3) torch_data=torch.from_numpy(np_data) tensor2array=torch_data.numpy() data=[-1,-1,2,-2] tensor=torch.FloatTensor(data) data=[[1,2],[3,4]] tensor=torch.FloatTensor(data) print( 'numpy',np_data, ' torch',torch_data, ' tensor2array',tensor2array ) print(' ',torch.mean(tensor)) print(' numpy',np.matmul(data,data), ' torch',torch.mm(tensor,tensor))