From aa1fdb5f3b5d6c54c3bc44269482c31f4dfafe67 Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Sat, 11 Apr 2026 01:04:29 +0200
Subject: Expose dense shape

---
 tensor/dense.go | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'tensor')

diff --git a/tensor/dense.go b/tensor/dense.go
index 9a2deb6..da1a5a0 100644
--- a/tensor/dense.go
+++ b/tensor/dense.go
@@ -65,6 +65,14 @@ func NewDense[T any](shape []int, base []T) Dense[T] {
 	return out
 }
 
+func (d Dense[T]) Shape() []int {
+	shape := make([]int, len(d.shape))
+
+	copy(shape, d.shape)
+
+	return shape
+}
+
 func (d Dense[T]) Size() int {
 	n := 1
 
-- 
cgit v1.2.3