pytorch
db15d191 - Update NestedTensor add to support non identical striding for NT+NT (#97195)

Commit
1 year ago
Update NestedTensor add to support non identical striding for NT+NT (#97195) # Summary NestedTensors currenlty don't support non-identical strided addition. When accumulating grad it possible to try and accumulate a grad with different striding then the old var and there is no way to change this in user code. This is a solution.. probs should support strided addition for NT Pull Request resolved: https://github.com/pytorch/pytorch/pull/97195 Approved by: https://github.com/albanD, https://github.com/cpuhrsch
Author
Committer
Parents
Loading