【AI】《动手学-深度学习-PyTorch版》笔记（十六）：自定义网络层、保存/加载参数、使用GPU

AI学习目录汇总

1、自定义网络层

自定义网络层很简单，三步即可完成

• 继承类：nn.Module
• 定义初始化函数：__init__中定义需要初始化的代码
• 定义向前传播函数：forward

1.1 不带参数的网络层

1）定义网络层

import torch
import torch.nn.functional as F
from torch import nn

class CenteredLayer(nn.Module)