flatten deeplearn.js

flatten.ts

/**
 * Flattens an input tensor.
 * @param inputs Tensor input
 */

function flatten(inputs: Tensor) {
  return g.reshape(inputs, (() => {
    let i = 1;
    inputs.shape.forEach((val) => { i *= val });
    return [i];
  })());
}