viable/strict/1788205825: [BE] Extend dropout support to complex numbers (#195373)

发生了什么

Summary nn.functional.dropout raised NotImplementedError for complex inputs on CPU (and CUDA), while silently "working" on MPS. Root cause: _dropout_impl allocated the Bernoulli mask ( noise ) in the input's dtype and called bernoulli_ on it. For complex inputs that requires a complex bernoulli_ , which CPU and CUDA do not implement (both dispatch via AT_DISPATCH_ALL_TYPES_AND3 , no complex). MPS only accepted it be…

摘要按规则整理自下方来源原文

来源