Pytorch的入门

• 1: 安装pip 包, 指定阿里云的镜像

  • pip install -i https://mirrors.aliyun.com/pypi/simple torch

• 2: 计算机中的随机数都是伪随机数

  • 需要设定随机种子.

• 3: 张量运算

  • data.add()
  • dada.sub()
  • data.mul()
  • data.div()
  • data.neg()
  • 都是矩阵中的每一个元素都要做对应的计算.

• 4: 阿达母积 & 点积 (矩阵乘法)

  data1 = [[1, 2]
           [3, 4]]
  data2 = [[5, 6]
           [7, 8]]

  data = torch.mul(data1, data2)
  等效于
  data = data1 * data2

  print(data)

  [[5, 12]
   [21, 32]]

  ----------------------------

  [[19, 22]
   [43, 50]]

• 矩阵的广播机制

  torch.matmul(torch.randn(3, 4, 5), torch.randn(5, 4)).shape

  广播之前: [3,4,5] --- [5,4]
  广播之后: [3,4,5] --- [3,5,4]
  矩阵乘法: [3,4,4]

  [5,4]  ---  [3,4,5]
  [3,5,4] --- [3,4,5]
  [3,5,5]

• torch.cuda.is_available() 判断服务器, 电脑是否支持cuda (GPU)