Create diagonal matrices from batch

[Samue1]

I have a batch

x = torch.rand(size=(M, N))

and want to create for each of the M inputs a diagonal matrix with dimensions N x N such that the output has dimensions M x N x N. How can I do that? If I pass x to torch.diag I get a one-dimensional output.

Any idea what I do wrong?

[KFrank]

Hi Samuel!

@Samue1

x = torch.rand(size=(M, N))

and want to create for each of the M inputs a diagonal matrix

Try:

torch.diag_embed (torch.rand (size = (M, N)))

Best.

K. Frank