Update TensorExtensions.cs

kasperk81 · kasperk81 · commit ff7870fce879 · 2024-11-10T18:05:17.000+02:00
diff --git a/src/libraries/System.Numerics.Tensors/src/System/Numerics/Tensors/netcore/TensorExtensions.cs b/src/libraries/System.Numerics.Tensors/src/System/Numerics/Tensors/netcore/TensorExtensions.cs
@@ -3500,7 +3500,7 @@ public static ref readonly TensorSpan<T> StackAlongDimension<T>(scoped ReadOnlyS
             Tensor<T>[] outputs = new Tensor<T>[tensors.Length];
             for (int i = 0; i < tensors.Length; i++)
             {
-                outputs[i] = Tensor.Unsqueeze(tensors[0], dimension);
+                outputs[i] = Tensor.Unsqueeze(tensors[i], dimension);
             }
             return ref Tensor.ConcatenateOnDimension<T>(dimension, tensors, destination);
         }

Original file line number	Diff line number	Diff line change
`@@ -3500,7 +3500,7 @@ public static ref readonly TensorSpan<T> StackAlongDimension<T>(scoped ReadOnlyS`
`3500`	`3500`	`Tensor<T>[] outputs = new Tensor<T>[tensors.Length];`
`3501`	`3501`	`for (int i = 0; i < tensors.Length; i++)`
`3502`	`3502`	`{`
`3503`		`- outputs[i] = Tensor.Unsqueeze(tensors[0], dimension);`
	`3503`	`+ outputs[i] = Tensor.Unsqueeze(tensors[i], dimension);`
`3504`	`3504`	`}`
`3505`	`3505`	`return ref Tensor.ConcatenateOnDimension<T>(dimension, tensors, destination);`
`3506`	`3506`	`}`