return fill for diagview uniformdiagtensor

kmp5VT · kmp5VT · commit a3c13c609cad · 2024-06-13T19:53:56.000-04:00
diff --git a/NDTensors/src/diag/diagtensor.jl b/NDTensors/src/diag/diagtensor.jl
@@ -36,7 +36,7 @@ function DiagonalArrays.diagview(T::NonuniformDiagTensor)
 end
 
 function DiagonalArrays.diagview(T::UniformDiagTensor)
-  return array(T)
+  return fill(getdiagindex(T, 1), diaglength(T))
 end
 
 ## Should this go in dense.jl or here since its related to diag?
