Method forward

imperative/python/megengine/module/activation.py:42–43

(self, inputs)

        self.axis = axis

    def forward(self, inputs):
        return softmax(inputs, self.axis)

    def _module_info_string(self) -> str:
        return "axis={axis}".format(axis=self.axis)