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

Fixing up some minor doc nits for tf.lookup

Browse Source 
PiperOrigin-RevId: 343916994
Change-Id: I6cb893e41451b6d9927511575198bec5f8dd8919
This commit is contained in:
Rohan Jain 2020-11-23 13:27:08 -08:00 committed by TensorFlower Gardener
parent 611b33942c
commit 417abb72af
1 changed files with 2 additions and 3 deletions

5
tensorflow/python/ops/lookup_ops.py
View File

@ -458,7 +458,7 @@ class DatasetInitializer(TableInitializerBase):
"""
def __init__(self, dataset):
"""Creates a table initializser from a `tf.data.Dataset`.
"""Creates a table initializer from a `tf.data.Dataset`.
Args:
dataset: A `tf.data.Dataset` object that produces tuples of scalars. The
@ -633,8 +633,7 @@ class TextFileInitializer(TableInitializerBase):
>>> init = tf.lookup.TextFileInitializer(
... filename=f.name,
... key_dtype=tf.string, key_index=tf.lookup.TextFileIndex.WHOLE_LINE,
... value_dtype=tf.int64, value_index=tf.lookup.TextFileIndex.LINE_NUMBER,
... delimiter=" ")
... value_dtype=tf.int64, value_index=tf.lookup.TextFileIndex.LINE_NUMBER)
>>> table = tf.lookup.StaticHashTable(init, -1)
>>> table.lookup(tf.constant('palmer 30')).numpy()
2