Track tensors emanating from `tf.data.Dataset.enumerate()`

Open khatchad opened this issue 1 year ago • 2 comments

Since https://github.com/wala/ML/pull/132, we can enumerate() over datasets and track the tensors. However, the TF Data API has a method tf.data.Dataset.enumerate() that does is similar to Python's enumerate(). But, we don't track tensors coming out of that API yet.

Related to #164.

Mar 11 '24 19:03 khatchad

zip() looks very similar.

Mar 11 '24 19:03 khatchad

Seems to work when using Jython 3.

Mar 12 '24 14:03 khatchad