Explan the input and weight Vector's Shape in Cpp

Let's see the following code (From Microsoft's AI-System Repo)

In the following code, for example Batch size is 4, Neuro Number is 5, Synapse each Neuron is 10, is:

M = input.size(0) = 4 = Bathsize,

K = input.size(1) = 10 = InputDim,

N = weights.size(0) = 5 = NeuroNum,

weights.size(1) = 10 = SynapseEachNeuron?

Let's See the Experiment Result:

My Guess is Right!😎