Bugfix in concatenating gradient batches in integrated gradients colab.

[mhavasi]

tf.stack only accounted for the last batch of gradients, instead of collecting the gradients from all batches. All but the last batch were discarded. The correct implementation is to concatenate the gradient batches by the first axis. This way, all the gradients are correctly accounted for.

Tested the changes by running the notebook in googe colab.

[github-actions]

Preview

Preview and run these notebook edits with Google Colab:

• site/en/tutorials/interpretability/integrated_gradients.ipynb

Rendered notebook diffs available on ReviewNB.com.

Format and style

Use the TensorFlow docs notebook tools to format for consistent source diffs and lint for style:

$ python3 -m pip install -U --user git+https://github.com/tensorflow/docs

$ python3 -m tensorflow_docs.tools.nbfmt notebook.ipynb

$ python3 -m tensorflow_docs.tools.nblint --arg=repo:tensorflow/docs notebook.ipynb

If commits are added to the pull request, synchronize your local branch: git pull origin patch-1

[MarkDaoust]

Oops, that was my mistake. Thanks!