晋江今天新增疫情最新消息_外网常用网站_媒体平台推广_互联网公司排名2021

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 * data2print(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)