How can I reshape Datafrom BigBatchSize to Just right BatchSize

I want to Inspection picture in Libtorch.
but my data is too big batches to Inspect on GPU like [9857 , 3, 64, 64]
so I want to split data like [5000 , 3, 64, 64] and [4857 , 3, 64, 64]

I tryed “reshape” command but batch count is different

How can I split data by different batch count


sorry I found “split” command.

so this topic is solved