![]() |
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 .
|