capsule-networks
capsule-networks copied to clipboard
Why the dimensions of weights are dependent on batch_size?
In ANN and CNN, the weights are not dependent on batch_size. But here (wij and bij) it is tiled according to batch_size and that number of weights will be trained.
What if we want to input a single image and get its classification. The whole model needs to be reconstructed and retrained for that batch_size=1?