Python day48.
@浙大疏锦行 Python day48
生成随机张量可以用于权重的初始化以及模拟输入,常用的有如下部分
- torch.randn():生成标准正态分布的张量
- torch.rand():生成[0,1)范围内均匀分布的随机数
- torch.normal():生成指定均值和标准差的正态分布
- 。。。
- 针对常见的张量我们需要考虑的主要有四部分:形状、数据类型、是否计算梯度以及使用的设备
广播机制:两个形状不同的张量进行运算时,PyTorch会自动处理适应操作,补全规则如下:
- 从第一个维度开始进行比较,遇见不同的如果为1则扩展,否则报错
- 若一个张量的维度少于另一个,则左侧直0接补1直至维度数匹配
- 此外,扩展是逻辑上的扩展,实际上未复制数据