BlackBird123:

So from this I am learning that the number of units in the Keras definition is the second number in the Torch layer definition.


Yes, the specified units in Dense are corresponding to the out_features in PyTorch.



 BlackBird123:

I’m still not certain how to reat the Input() definition from Keras. Is it a linear layer like I have it?


You don’t need to specify input layers or any placeholders in PyTorch and can directly pass the input tensors to the model.



 BlackBird123:

Also, do you think the he_uniform_initialization matters?


Yes, initialization can matter but it also depends on the model.