tf.sets.union

Compute set union of elements in last dimension of a and b.

All but the last dimension of a and b must match.

Example:

  import tensorflow as tf
  import collections

  # [[{1, 2}, {3}], [{4}, {5, 6}]]
  a = collections.OrderedDict([
      ((0, 0, 0), 1),
      ((0, 0, 1), 2),
      ((0