This is true for function ```lwta()``` but not for ```LWTA.fprop()``` method. ```lwta()``` is applied on ```p``` which is wx+b.