'Given a partitioning, slices, of an activation buffer, x, determine which axis to slice
along depending on whether x is a sequential tensor or not.
Arguments:
x:
slices:
Returns:'

def get_partitions(self, x, slices):

    if (x.shape[(-1)] != self.be.bsz):
        return [x[:, sl] for sl in slices]
    else:
        return [x[sl] for sl in slices]
