nsl.lib.apply_feature_mask

Applies a feature mask on features if the feature_mask is not None.

features A dense tensor representing features.
feature_mask A dense tensor with values in [0, 1] and a broadcastable shape to features. If not set, or set to None, the features are returned unchanged.

A dense tensor having the same shape as features.