[tensor] reorder the sets toTensor to use len of produced seq

mratsim · Jun 16, 2024 · ad4cbc2 · ad4cbc2
1 parent 5b3c84c
commit ad4cbc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/arraymancer/laser/tensor/initialization.nim b/src/arraymancer/laser/tensor/initialization.nim
@@ -284,8 +284,8 @@ proc toTensor*[T](a: HashSet[T] | OrderedSet[T]): Tensor[T] =
   ## Result:
   ##      - A Tensor of the same shape
   var shape = MetaData()
-  shape.add(a.len)
   let data = toSeq(a)
+  shape.add(data.len)
   result = toTensor(data, shape)
 
 proc toTensor*[T; U](a: HashSet[T] | OrderedSet[T], typ: typedesc[U]): Tensor[U] {.inline.} =