Searched defs:boolean_mask (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
H A Dops.py1213 def boolean_mask(labeled_tensor, mask, name=None): function
1216 Unlike `tf.boolean_mask`, this currently only works on 1-dimensional masks.
1237 "LabeledTensor's boolean_mask currently only supports 1D masks")
1243 op = array_ops.boolean_mask(labeled_tensor.tensor, mask.tensor, name=scope)
/external/tensorflow/tensorflow/contrib/labeled_tensor/
H A D__init__.py121 boolean_mask = _ops.boolean_mask variable
/external/tensorflow/tensorflow/python/ops/
H A Darray_ops.py67 @@boolean_mask
1179 @tf_export("boolean_mask")
1180 def boolean_mask(tensor, mask, name="boolean_mask", axis=None): function
1187 boolean_mask(tensor, mask) # [0, 2]
1192 `boolean_mask(tensor, mask)[i, j1,...,jd] = tensor[i1,...,iK,j1,...,jd]`
1219 boolean_mask(tensor, mask) # [[1, 2], [5, 6]]

Completed in 168 milliseconds