pytorch中一些常用的模块中的函数(持续更新)
import torch # 神经网络用
import torch.nn.functional as F # 激活函数使用
import torch.utils.data as Data # 批处理使用
常用函数
torch.nn.Linear() # 线性层
torch.nn.MSELoss() # 均方差损失函数(回归用)
torch.nn.CrossEntropyLoss() # # 交叉...
原创
2019-05-07 16:36:02 ·
3359 阅读 ·
0 评论