What would be the best way to make an identity matrix directly on GPU?
I couldn’t find a torch.cuda.eye() factory function.
I am currently simply making a
FloatTensor, then filling it with zeros with
.fill_(0) and finally
torch.masked_scatter_() to put ones on the diagonal.
Any faster/more elegant way to do this?