pytorch中的nn.Module和nn.functional模块

一、nn.Module类

通过继承 nn.Module 转换成为类的实现形式,并直接封装在 nn 模块下,其中包含了常见的卷积层、池化层、以及损失函数等,下面将会介绍该模块下的一些函数,通常我们会import torch.nn as nn,以下都基于nn来表述:
1.激活函数:
nn.Relu
nn.Sigmoid
nn.Tanh
nn.Softmax
2.模型层:
nn.Linear
nn.Conv2d
nn.MaxPool2d
nn.Dropout2d
nn.Embedding
3.损失函数:
nn.BCELoss
nn.MSELoss
nn.CrossEntropyLoss

二、nn.functional模块

该模块的功能和nn模块差不多,包括损失函数、网络层之类的,通过该模块去调用相应的函数其实也是转到了nn模块下,所以该方法不常用,我也不常用,所以不做具体介绍。

之后会通过实战实例来详细介绍这写模块的函数调用方法,毕竟手撕代码比理论学习来得更加清晰,更加迅速。

更多技术欢迎加入交流:320297153

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值