subtitle
Pytorch函数总结-模型相关函数 Pytorch函数总结-模型相关函数
名字 功能 参数 定义 备注1 备注2 nn.Linear 对于输入数据,做一个线性转换 “输入特征数目 输出特征数目 是否需要偏置” nn.Conv2d 对于输入数据,做2D的卷积运算 “in_channels 输
Pytorch函数总结-GPU相关函数 Pytorch函数总结-GPU相关函数
名字 功能 参数 定义 备注1 备注2 torch.cuda.is_available() 查看gpu是否可用 torch.cuda.device_count() gpu数量 torch.cuda.current_d
Pytorch函数总结-常用函数 Pytorch函数总结-常用函数
一、Pytorch函数总结 函数 功能 参数 定义 备注 示例 示例 torch.mul 对位相乘 两个矩阵 或者一个矩阵一个标量 torch.mul(input, other, *, out=None) → Tensor “
Cha3 深度学习基础-Pytorch 函数总结 Cha3 深度学习基础-Pytorch 函数总结
一、Pytorch函数总结 函数 功能 参数 定义 备注 示例 示例 torch.mul 对位相乘 两个矩阵 或者一个矩阵一个标量 torch.mul(input, other, *, out=None) → Tensor “