tf.data: function fails tries to convert to numpy array?

Question

I\'m trying to build a tf.data pipeline, ultimately to compute skipgrams, but I get an error

NotImplementedError: Cannot convert a symbolic Tensor (c