mlx.nn.Sequential

目录

mlx.nn.Sequential#

class Sequential(*modules)#

一个按顺序调用传入的可调用对象的层。

我们可以将模块或普通的可调用对象传递给 Sequential 模块。如果我们的函数具有可学习参数,则应将其实现为 nn.Module 实例。

参数:

modules (元组,包含可调用对象) – 按顺序调用的模块

方法