pub fn ce_bwd(probs: &Tensor, ids: &Tensor, scale: f32) -> Result<Tensor>
Cross-entropy backward: dlogits = (probs - onehot(ids)) * scale.