PyTorch 等效於 Keras 順序 model

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

PyTorch 等效：

nn.Sequential(OrderedDict([
    ('1', nn.Flatten()),
    ('2', nn.Linear(28 * 28, 128)),
    ('3', nn.ReLU()),
    ('4', nn.Linear(128, 10))
]))