Still having troubles with neural net shapes

Are you saying it’s as simple as switching the weights in the shapes? So on weights_2_3 instead of shape being (64, 128) it should be (128,64)?