experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Changed tensor pseudo-code so that the example is runnable

Browse Source
This commit is contained in:
Karl Weinmeister 2019-02-01 19:57:33 -06:00 committed by GitHub
parent 5679c6f03a
commit 131270d780
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/python/data/ops/dataset_ops.py
View File

@ -1101,18 +1101,15 @@ class DatasetV2(object):
"""Filters this dataset according to `predicate`. """Filters this dataset according to `predicate`.
```python ```python
# NOTE: The following examples use `{ ... }` to represent the d = tf.data.Dataset.from_tensor_slices([1, 2, 3])
# contents of a dataset.
a = { 1, 2, 3 }
b = { 4, 5, 6, 7 }
a.filter(lambda x: x < 3) == { 1, 2 } d = d.filter(lambda x: x < 3) # [1, 2]
# `tf.math.equal(x, y)` is required for equality comparison # `tf.math.equal(x, y)` is required for equality comparison
def filter_fn(x): def filter_fn(x):
return tf.math.equal(x, 4) return tf.math.equal(x, 1)
b.filter(filter_fn) == { 4 } d = d.filter(filter_fn) # [1]
``` ```
Args: Args: