Torch.diag for batches?

1

How to use torch.diag for batches?

2

I’d use torch.diagonal and torch.diag_embed instead.

Best regards

Thomas