trunk/736985d0de6c552c8fe2324b101cc94bd21b245b: [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…

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

来源