There was an error while loading. Please reload this page.
Also align with implementations like tensorflow.
Working on a PR for this.