(self, x, slices)
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:
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.
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 .
def get_partitions(self, x, slices):
        """
        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:

        """
        if x.shape[-1] != self.be.bsz:  # This is the sequential case
            return [x[:, sl] for sl in slices]
        else:
            return [x[sl] for sl in slices]
def get_partitions ( self , x , slices ) : if x . shape [ - 1 ] != self . be . bsz : # This is the sequential case return [ x [ : , sl ] for sl in slices ] else : return [ x [ sl ] for sl in slices ]