In a similar vein, if I do use ShapeProp to follow dummy Tensors through the graph, is there a way to keep track of which dimensions “are” or derive from the batch dimension? (Short of picking a unique, and thus likely large, number?)