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)