mask_c = np.load( path.join(config.ref_log_dir, config.ref_scene, 'diff_rgb_rays.npy')) diff_origins = torch.tensor(mask_c[:,0:3]) diff_directions = torch.tensor(mask ...