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