Class ov::op::v1::Softmax#
-
class Softmax : public ov::op::Op#
Softmax operation.
Public Functions
-
Softmax(const Output<Node> &arg, const size_t axis = 1)#
Constructs a softmax operation.
Output
[d0, ...]- Parameters:
arg – Node that produces the first input tensor.
[d0, ...]axis – The axis position (0-based) on which to calculate the softmax.
-
virtual void validate_and_infer_types() override#
Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.
Throws if the node is invalid.
-
virtual bool has_evaluate() const override#
Allows to get information about availability of evaluate method for the current operation.
-
Softmax(const Output<Node> &arg, const size_t axis = 1)#