From 456f8218a2d167385ae5570b04d45fb1d73befa3 Mon Sep 17 00:00:00 2001
From: Jonas Knobloch <jonas.knobloch@t-online.de>
Date: Fri, 10 Apr 2026 23:04:08 +0200
Subject: Construct dense with base

---
 tensor/dense.go | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

(limited to 'tensor/dense.go')

diff --git a/tensor/dense.go b/tensor/dense.go
index 723cba5..9a2deb6 100644
--- a/tensor/dense.go
+++ b/tensor/dense.go
@@ -11,7 +11,7 @@ type Dense[T any] struct {
 	strides []int
 }
 
-func NewDense[T any](shape []int) Dense[T] {
+func NewDense[T any](shape []int, base []T) Dense[T] {
 	r := len(shape)
 
 	if r == 0 {
@@ -33,6 +33,14 @@ func NewDense[T any](shape []int) Dense[T] {
 		n *= shape[dim]
 	}
 
+	if base == nil {
+		base = make([]T, n)
+	}
+
+	if len(base) != n {
+		panic("unexpected base length")
+	}
+
 	strides := make([]int, len(shape))
 
 	for dim := range slices.Backward(strides) {
@@ -46,7 +54,7 @@ func NewDense[T any](shape []int) Dense[T] {
 	}
 
 	out := Dense[T]{
-		base:    make([]T, n),
+		base:    base,
 		offset:  0,
 		shape:   make([]int, len(shape)),
 		strides: strides,
@@ -107,14 +115,14 @@ func (d Dense[T]) IsContiguous() bool {
 
 func (d Dense[T]) Contiguous() Dense[T] {
 	if d.IsContiguous() {
-		out := NewDense[T](d.shape)
+		out := NewDense[T](d.shape, nil)
 
 		copy(out.base, d.base[d.offset:d.offset+d.Size()])
 
 		return out
 	}
 
-	out := NewDense[T](d.shape)
+	out := NewDense[T](d.shape, nil)
 
 	idxs := make([]int, d.Rank())
 
-- 
cgit v1.3.1