AI学习目录汇总

1、自定义网络层

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

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

1.1 不带参数的网络层

1）定义网络层

import torch
import torch.nn.functional as F
from torch import nnclass CenteredLayer(nn.Module)