Function getLayerSummary

tfjs-vis/src/show/model.ts:151–166

(layer: Layer)

 * Gets summary information/metadata about a layer.
 */
function getLayerSummary(layer: Layer): LayerSummary {
  let outputShape: string;
  if (Array.isArray(layer.outputShape[0])) {
    const shapes = (layer.outputShape as number[][]).map(s => formatShape(s));
    outputShape = `[${shapes.join(', ')}]`;
  } else {
    outputShape = formatShape(layer.outputShape as number[]);
  }

  return {
    name: layer.name,
    trainable: layer.trainable,
    parameters: layer.countParams(),
    outputShape,
  };
}

interface LayerSummary {
  name: string;