could use tuple directly, i.e.,
    _reshape = lx_outer_shape +  x_out.shape[-2:]
    return x_out.reshape(_reshape)