m0_51579041的博客

python index_put_ 用法说明

首先，torch.cat是在已有的维度上拼接，比如两个形状是(2,3)的张量，用cat在dim=0的话，就会变成(4,3)，而dim=1的话就是(2,6)。而torch.stack则会新建一个维度，比如两个(3,4)的张量，用stack在dim=0的话，结果变成(2,3,4)。比如A是(2,3)，B是(2,3)，cat之后在dim=0变成(4,3)，dim=1变成(2,6)。而stack的话，结果会是(2,2,3)或者(2,3,2)，取决于dim参数。比如用cat的时候，除了拼接的维度，其他维度必须相同。

torch.clip 裁剪函数

add_module

怎么检查cuda是否安装成功（以及查看cuda的安装位置）

torch.cumsum 前缀和

torch.sort()和torch.argsort()函数解析

torch.linspace() torch.arange() torch.stack()

torch维度操作

图像归一化和反归一化

通过tqdm实现进度条打印训练过程信息

torch.Tensor 和 torch.from_numpy

torch.mean()

DataParallel和DistributedDataParallel

torch中的optimizer.zero_grad(), loss.backward(), optimizer.step()