o پi@s@dZddlZddlmmZddlmZGdddejZdS)z% Linear layer (alternate definition) N)nnc@s&eZdZdZdejdejfddZdS)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. inputreturncCsVtjr"|jdur|jj|jdnd}tj||jj|jd|dSt||j|jS)N)dtype)bias) torchjit is_scriptingrtorFlinearweight)selfrrrF/home/ubuntu/.local/lib/python3.10/site-packages/timm/layers/linear.pyforwards zLinear.forwardN)__name__ __module__ __qualname____doc__rTensorrrrrrrsr)rrtorch.nn.functionalr functionalr rrrrrs