From 95c795c7ebdddc536a610b1a6a7de45777ae2ca0 Mon Sep 17 00:00:00 2001 From: Benjamin Meyer Date: Sat, 15 Aug 2026 09:13:01 +0200 Subject: [PATCH] bugfix: take has wrong output type. --- .../main/scala/dimwit/tensor/tensorops/StructuralOps.scala | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/src/main/scala/dimwit/tensor/tensorops/StructuralOps.scala b/core/src/main/scala/dimwit/tensor/tensorops/StructuralOps.scala index a88d305..a5042e3 100644 --- a/core/src/main/scala/dimwit/tensor/tensorops/StructuralOps.scala +++ b/core/src/main/scala/dimwit/tensor/tensorops/StructuralOps.scala @@ -653,9 +653,9 @@ object StructuralOps: )( indices: Tensor1[L2, Int32] )(using - ev: AxisRemover[T, L1], - labels: Labels[ev.RemainingAxes] - ): Tensor[Tuple.Concat[Tuple1[L2], ev.RemainingAxes], V] = + ev: AxisReplacer[T, L1, L2], + labels: Labels[ev.NewShape] + ): Tensor[ev.NewShape, V] = val result = Jax.jnp.take(tensor.jaxValue, indices.jaxValue, axis = ev.index) Tensor(result)