pytorch
1c0f0b33 - Update `amax/amin/norm/count_nonzero` signatures with `int[*]? dim` (#83300)

Commit
2 years ago
Update `amax/amin/norm/count_nonzero` signatures with `int[*]? dim` (#83300) Changes `dim` arg to use `int[*]?` type for the following functions in `native_funcitons.yaml`: * `amax` * `amin` * `norm` * `frobenius_norm` * `native_norm` * `count_nonzero` Part of #29137 Pull Request resolved: https://github.com/pytorch/pytorch/pull/83300 Approved by: https://github.com/ngimel, https://github.com/albanD, https://github.com/kulinseth
Author
Committer
Parents
Loading