mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Fix typo: 'complext' (#160335)
minor fix for a typo: `complext` to `complex` Pull Request resolved: https://github.com/pytorch/pytorch/pull/160335 Approved by: https://github.com/Skylion007
This commit is contained in:
committed by
PyTorch MergeBot
parent
037c43d3b2
commit
4651aaac47
@ -139,7 +139,7 @@ struct Dist {
|
||||
static inline data_t map(const data_t& diff, const data_t& p) { return diff; }
|
||||
static inline data_t red(const data_t& agg, const data_t& up) { return max(agg, up); }
|
||||
static inline scalar_t finish(const scalar_t agg, const scalar_t p) { return agg; }
|
||||
// TODO This backward pass uses a very complext expression to compute (diff
|
||||
// TODO This backward pass uses a very complex expression to compute (diff
|
||||
// == dist) that could be much faster if using SSE instructions.
|
||||
static inline Vec backward(const Vec& diff, const scalar_t grad, const scalar_t dist, const Vec& p) { return Vec(grad) * sign(diff) * (Vec(1) - vec::minimum(Vec(1), (diff.abs() - Vec(dist)).abs().ceil())); }
|
||||
};
|
||||
|
Reference in New Issue
Block a user