Missed TensorFlow World? Check out the recap. Learn more

nsl.lib.apply_feature_mask

View source on GitHub

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

nsl.lib.apply_feature_mask(
    features,
    feature_mask=None
)

Args:

  • 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.

Returns:

A dense tensor having the same shape as features.