How will we be able to do split across batches? So if my tensor changes to 2D for example.
A = [[1.3, 0.0, 0.6, 0.7], [1.1,1.2, 0.0, 0.0]]
My length of split will be then [[1, 1, 2], [2, 1, 1]]. torch.split() only takes the list and not list of list so is it a new function that I will have to make?
I checked your feature request as well on github but the split size is changing across a dimension for me.