Google I/O returns May 18-20! Reserve space and build your schedule Register now


Strips graph neighbor features from a feature dictionary.

features Dictionary of tensors mapping feature names to tensors. This dictionary includes sample features but may or may not include corresponding neighbor features for each sample feature.
neighbor_config An instance of nsl.configs.GraphNeighborConfig.

A dictionary mapping only sample feature names to tensors. Neighbor features in the input are not included.