darabos's picture
Do not track tensor shapes. Much simpler!
612802c