def METHOD_NAME(self, other):
    *batch_shape, num_rows, num_cols = other.shape
    batch_shape = list(batch_shape)
    batch_shape.append(num_rows // self.num_blocks)
    other = other.view(*batch_shape, self.num_blocks, num_cols)
    other = other.transpose(-2, -3).contiguous()
    return other