machine-learning - 为什么 Trax 会自动在子图层上创建串行图层？

Question

我在Trax（Google 的深度学习库）中实现了一个串行层。尽管已经声明了一个，为什么还要创建一个额外的 Serial 层？

下面是代码。

model = tl.Serial(
        tl.Dense(n_units=512),
        tl.Relu()
        )
print(model)

输出是：

Serial[
  Dense_512
  Serial[
     Relu
  ]

]

虽然它应该是

Serial[
  Dense_512
  Relu
]