Calculates an element-wise squared difference between two tensors. More...
#include <squared_difference.hpp>
Public Member Functions | |
const NodeTypeInfo & | get_type_info () const override |
SquaredDifference (const Output< Node > &x1, const Output< Node > &x2, const AutoBroadcastSpec &auto_broadcast=AutoBroadcastType::NUMPY) | |
Constructs the squared difference operation. More... | |
bool | visit_attributes (AttributeVisitor &visitor) override |
virtual OutputVector | decompose_op () const override |
virtual std::shared_ptr< Node > | clone_with_new_inputs (const OutputVector &new_args) const override |
const AutoBroadcastSpec & | get_autob () const override |
void | set_autob (const AutoBroadcastSpec &auto_broadcast) |
Static Public Attributes | |
static constexpr NodeTypeInfo | type_info {"SquaredDifference", 0} |
Calculates an element-wise squared difference between two tensors.
y[i] = (x1[i] - x2[i])^2
ngraph::op::v0::SquaredDifference::SquaredDifference | ( | const Output< Node > & | x1, |
const Output< Node > & | x2, | ||
const AutoBroadcastSpec & | auto_broadcast = AutoBroadcastType::NUMPY |
||
) |
Constructs the squared difference operation.
x1 | First input tensor |
x2 | Second input tensor |
auto_broadcast | Auto broadcast specification |