viz.renderers.FeatureMapGridRenderer

class FeatureMapGridRenderer

GPU-accelerated feature map grid renderer.

Examples

const renderer = new FeatureMapGridRenderer();

// Visualize all 64 feature maps from a conv layer
// activations shape: [64, 28, 28]
const target = await renderer.renderAutoRange(activations, {
  cellSize: 32,
  cols: 8,
});