Keras 顺序模型的 PyTorch 等价物

Question

如何在PyTorch中完美复制这个Keras时序网络？

model = tf.keras.models.Sequential([
    tf.keras.layers.Flatten(input_shape=(28, 28)),
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(10)
])

Answer 1

这是适用于这种情况的片段：

    model_torch = nn.Sequential(
         nn.Flatten(), 
         nn.Linear(28*28, 128), 
         nn.ReLU(), 
         nn.Linear(128, 10), 
    )