| 41 | class DropoutModule : public Module { |
| 42 | public: |
| 43 | DropoutModule(const float dropRatio) { |
| 44 | mDropRatio = dropRatio; |
| 45 | setType("Dropout"); |
| 46 | } |
| 47 | |
| 48 | virtual std::vector<Express::VARP> onForward(const std::vector<Express::VARP>& inputs) override { |
| 49 | Express::VARP x = inputs[0]; |