Original Paper | arXiv |
Original Source | GitHub |
This code was written based off the original source in the Deep Mind repository. Validation includes ensuring numerically correct output is produced for same weight values.
Notes:
InvertableNorm
was created to match theNormalizer
from the original modelunsorted_segsum
implements TensorFlow's UnsortedSegmentSum.