size for MPSMatrix

Author: christiangnrd

lib/mps/matrix.jl

@@ -119,6 +119,13 @@ function MPSMatrix(arr::MtlArray{T,3}) where T
     return MPSMatrix(arr, desc, offset)
 end
 
+function Base.size(mat::MPS.MPSMatrix)
+    if mat.matrices > 1
+        return Int64.((mat.rows, mat.columns, mat.matrices))
+    else
+        return Int64.((mat.rows, mat.columns))
+    end
+end
 
 ## matrix multiplication