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
.
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
.