人工智能基础-pytorch篇

Pytorch

1. 创建tensor

• 使用numpy、list创建

  a = torch.tensor(np.array([1,2,3,4])
  b = torch.tensor([1,2,3,4])
  

• 使用empty、zero、ones初始化

  a = torch.empty(3, 5)
  b = torch.zeros(2, 5)
  c = torch.new_ones(4, 8, 9)
  

• 使用随机初始化

  a = torch.rand(2, 5)
  b = torch.randn(3, 5)
  

2. tensor的一些属性

• 查看tensor维度

  a.ndim
  

• 查看tensor形状

  a.shape
  

• 查看tensor的数据类型

  a.dtype
  

• 查看tensor的设备

  a.device
  

3. tensor的计算

• 数学计算

  # 逐元素计算
  a + b
  a - b
  a * b
  a / b
  a % b
  a // b
  
  # 矩阵点乘
  a.dot(b)
  
  #求平均值、最大值、最小值和求和
  a.mean()   # 元素类型为float
  a.max()
  a.min()
  a.sum()
  

• tensor维度变换

  a.view(2,3)
  

• 修改tensor的设备

  a.to("cuda")