Transform data to (batch, (data, data, features))

[ranyishere]

Reformat the data such that the persistent diagram takes in a batch size and a flattened numpy array of the image.