Merge "Add heatmap visualizations and video saving for intermediate tensors."