Fix tensor visualization.