[NDTensors] Define denseblocks on Dense (#1677)

Author: mtfishman

NDTensors/Project.toml

@@ -1,7 +1,7 @@
 name = "NDTensors"
 uuid = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
 authors = ["Matthew Fishman <[email protected]>"]
-version = "0.4.14"
+version = "0.4.15"
 
 [deps]
 Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"

NDTensors/src/dense/densetensor.jl

@@ -75,6 +75,8 @@ end
 # Useful for using Base Array functions
 array(T::DenseTensor) = convert(Array, T)
 
+denseblocks(T::DenseTensor) = T
+
 function diagview(T::DenseTensor)
     return diagview(array(T))
 end

NDTensors/test/test_dense.jl

@@ -1,5 +1,6 @@
 @eval module $(gensym())
 using NDTensors
+using NDTensors: denseblocks
 using NDTensors.MetalExtensions: mtl
 using Test: @testset, @test, @test_throws, @test_broken
 using GPUArraysCore: @allowscalar
@@ -30,6 +31,9 @@ NDTensors.dim(i::MyInd) = i.dim
             @test dims(A[2:3, 2:end]) == (2, 3)
             @test dims(A[3, 2:end]) == (3,)
 
+            @test dense(A) ≡ A
+            @test denseblocks(A) ≡ A
+
             randn!(A)
 
             @test ndims(A) == 2