Skip to content

Switch to `torch.diag`

torch.diag now supports complex tensors, so heron.utils.diag_cuda is redundant. This MR replaces all of the calls with torch.diag and removes heron.utils.diag_cuda.

Merge request reports