chore: adding more examples to the reshape method

machinelearnjs · Nov 29, 2018 · cbcf0b3 · cbcf0b3
1 parent 8d9b585
commit cbcf0b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/lib/ops/tensor_ops.ts b/src/lib/ops/tensor_ops.ts
@@ -93,6 +93,7 @@ export function validateMatrix2D(X: Type2DMatrix<any>): number[] {
  *
  * @example
  * reshape([1, 2, 3, 4, 5, 6], [2, 3]); // [[1, 2, 3], [4, 5, 6]]
+ * reshape([1, 2, 3, 4, 5, 6], [2, 3, 1]); // [[[1], [2], [3]], [[4], [5], [6]]]
  *
  * @param array - Target array
  * @param sizes - New array shape to resize into