for layer 1 = 64 neurons and layer2 = 128 neurons and layer3 = 256 neurons,the shape of activation[‘layer1’] is (100,64,32,32)
what is 100? what are the 32s?
also for activation[‘layer2’] the shape is  (100,128,16,16) and for activation['layer3] the shape is (100,256,8,8)