U ʹj@s@dZddlZddlmmZddlmZGdddejZdS)z% Linear layer (alternate definition) N)nnc@s$eZdZdZejejdddZdS)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. )inputreturncCsZtjrD|jdk r$|jj|jdnd}tj||jj|jd|dSt||j|jSdS)N)dtype)bias) torchZjitZ is_scriptingrtorFZlinearweight)selfrrr Q/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/layers/linear.pyforwards zLinear.forwardN)__name__ __module__ __qualname____doc__rZTensorrr r r rrsr)rrZtorch.nn.functionalrZ functionalr rr r r rs