- towhee.models.collaborative_experts.collaborative_experts.drop_nans(x, ind, validate_missing)¶
Remove nans, which we expect to find at missing indices.
x (torch.Tensor) – Features
ind (torch.Tensor) – Binary values denoting whether or not a given feature is present.
validate_missing (bool) – Whether to validate that the missing location contains a nan.
The features, with the missing values masked to zero.
- Return type: