add tensor visualization