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

Open khatchad opened this issue 1 year ago • 1 comments

          [`zip()`](https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/data/Dataset#zip) looks very similar.

Originally posted by @khatchad in https://github.com/wala/ML/issues/165#issuecomment-1989296987

Mar 11 '24 20:03 khatchad

Seems to work when using Jython 3.

Mar 12 '24 14:03 khatchad