Pytorch的入门

• 1: 安装pip 包, 指定阿里云的镜像
• pip install -i 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)