import numpy as np
# Create weights matching a Dense layer with 784 inputs and 128 units
np.save("fc1.weight.npy", np.random.randn(784, 128).astype(np.float32))
np.save("fc1.bias.npy", np.zeros(128, dtype=np.float32))
# Or bundle into an .npz archive
np.savez("model.npz",
**{"fc1.weight": np.random.randn(784, 128).astype(np.float32),
"fc1.bias": np.zeros(128, dtype=np.float32)})