introduction to pytorch [[[ ][ ]][[ ][ ]]]]in [ ] shape out[ ]torch size([ ]in [ ] torch flatten(ain [ ] out[ ]tensor([ ]in [ ] shape out[ ]torch size([ ]in [ ] torch flatten(astart_dim= in [ ] out[ ]tensor([ ]in [ ] shape out[ ]torch size([ ]in [ ] torch flatten(astart_dim= in [ ] out[ ]tensor([[ ][ ]]in [ ] shape out[ ]torch size([ ]