莫烦Pytorch系列教程2.1——torch vs Numpy

import torch
import numpy as np
np_data = np.arange(6).reshape((2,3)) #0到5，第一个轴为2，第二个轴为3
torch_data = torch.from_numpy(np_data)
tensor2arraay = torch_data.numpy()
print(
      '\nnumpy array', np_data, 
      '\ntorch_data', torch_data,
      '\ntensor2arraay',tensor2arraay,
      )

data =  [-1,-2,1,2]
tensor = torch.FloatTensor(data)
print(
      '\nabs'
      '\nnumpy_abs',np.abs(data),
      '\ntorch_abs',torch.abs(tensor),
      )

print(
      '\nsin',
      '\nnumpy_sin',np.sin(data),
      '\ntorch_sin',torch.sin(tensor),
      )

print(
      '\nmean'
      '\nnumpy_mean',np.mean(data),
      '\ntorch_mean',torch.mean(tensor),
      )

data1 = [[1,2],[3,4]]#列表
tensor1 = torch.FloatTensor(data1)#转换为浮点张量
print(
      '\nmatrix multiplication1',
      '\nnumpy_matrix_mul',np.matmul(data1,data1),
     
      '\ntorch_matrix_mul',torch.matmul(tensor1,tensor1)
      )

data2 = np.array(data1)#numpy数组
print(
      '\nmatrix multiplication2',
      '\nnumpy_matrix_mul',data2.dot(data1),
    
      )