0

我在 tensorflow keras 中构建了一个神经网络，最终得到了以下代码作为模型中的一个步骤：

enc = tfkl.Reshape((-1, 20,input_shape[1]))(input_layer)
encoder_output = []
for i in range(enc.shape[1]):
     o = encoder(enc[:, i, ...])
     encoder_output.append(o)
encoder_output = tf.stack(encoder_output, axis=1)

代码工作正常。然而，它的摘要相当长，而且通常不是很优雅。我对功能 API 和自定义层了解不多。但我想知道在这种情况下是否最好使用自定义层或构建功能 API 块。

谢谢

4

0 回答 0