def METHOD_NAME(self, other):
    other = other.transpose(-2, -3).contiguous()
    shape = list(other.shape)
    del shape[-2]
    shape[-2] *= self.num_blocks
    other = other.reshape(*shape)
    return other