NgVdZddlZddlmcmZddlmZGddejZdS)z% Linear layer (alternate definition) N)nnc6eZdZdZdejdejfdZdS)LinearzApplies a linear transformation to the incoming data: :math:`y = xA^T + b` Wraps torch.nn.Linear to support AMP + torchscript usage by manually casting weight & bias to input.dtype to work around an issue w/ torch.addmm in this use case. inputreturnc:tjr^|j |j|jnd}t j||j|j|St j||j|jS)N)dtype)bias) torchjit is_scriptingr tor Flinearweight)selfrr s N/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/layers/linear.pyforwardzLinear.forwards} 9 ! ! # # ;6:i6K49<> >#D#D4PPP P8E4; :: :N)__name__ __module__ __qualname____doc__r TensorrrrrrsD ;U\;el;;;;;;rr)rr torch.nn.functionalr functionalrrrrrrs{  ; ; ; ; ;RY ; ; ; ; ;r