swift
093f87ea - [AutoDiff] Optimize derivative for `Tensor.mean`. (#23523)

Commit

6 years ago

[AutoDiff] Optimize derivative for `Tensor.mean`. (#23523) Perform division by scalar before broadcasting. Idea by @ewconnell.

References

Author

dan-zheng

Committer

rxwei

Parents