viz.renderers.KernelGridRenderer

class KernelGridRenderer

GPU-accelerated CNN kernel visualization.

Examples

const renderer = new KernelGridRenderer();

// Get first conv layer weights [out_channels, in_channels, H, W]
const weights = model.conv1.weight;

// Visualize first input channel kernels
const kernels = weights.select(1, 0); // [out_channels, H, W]
renderer.render(kernels, {
  cellSize: 50,
  gridCols: 8,
});