MLP

class torchvision.ops.MLP(in_channels: int, hidden_channels: ~typing.List[int], norm_layer: ~typing.Optional[~typing.Callable[[...], ~torch.nn.modules.module.Module]] = None, activation_layer: ~typing.Optional[~typing.Callable[[...], ~torch.nn.modules.module.Module]] = <class 'torch.nn.modules.activation.ReLU'>, inplace: ~typing.Optional[bool] = None, bias: bool = True, dropout: float = 0.0)

此模块实现了多层感知机 (MLP) 模块。

参数:

• in_channels (int) – 输入通道数

• hidden_channels (List[int]) – 隐藏通道维度列表

• norm_layer (Callable[..., torch.nn.Module], optional) – 将堆叠在线性层之上的归一化层。如果 None，则不会使用此层。默认值：None

• activation_layer (Callable[..., torch.nn.Module], optional) – 激活函数，将堆叠在归一化层之上（如果不是 None），否则堆叠在线性层之上。如果 None，则不会使用此层。默认值：torch.nn.ReLU

• inplace (bool, optional) – 激活层的参数，可以选择就地执行操作。默认值为 None，它使用 activation_layer 和 Dropout 层的各自默认值。

• bias (bool) – 是否在线性层中使用偏置。默认值 True

• dropout (float) – Dropout 层的概率。默认值：0.0