Function inplace_attn_softmax_last_dim

pub fn inplace_attn_softmax_last_dim(
    xs: &mut Tensor,
    mask: &Tensor,
    scale: f32,
) -> Result<()>

Inplace equivalent of attn_softmax_last_dim