def subtract(left_node, right_node, name=None): # type: (NodeInput, NodeInput, str) -> Node """Return node which applies f(x) = A-B to the input nodes element-wise. :param left_node: The node providing data for left hand side of operator. :param right_node: The node providing data for right hand side of operator. :param name: The optional name for output node. :return: The new output node performing subtraction operation on both tensors element-wise. """ return Subtract(left_node, right_node)
def binary_op(op_str, a, b): if op_str == '+': return a + b elif op_str == 'Add': return Add(a, b) elif op_str == '-': return a - b elif op_str == 'Sub': return Subtract(a, b) elif op_str == '*': return a * b elif op_str == 'Mul': return Multiply(a, b) elif op_str == '/': return a / b elif op_str == 'Div': return Divide(a, b) elif op_str == 'Dot': return Dot(a, b) elif op_str == 'Equal': return Equal(a, b) elif op_str == 'Greater': return Greater(a, b) elif op_str == 'GreaterEq': return GreaterEq(a, b) elif op_str == 'Less': return Less(a, b) elif op_str == 'LessEq': return LessEq(a, b) elif op_str == 'Maximum': return Maximum(a, b) elif op_str == 'Minimum': return Minimum(a, b) elif op_str == 'NotEqual': return NotEqual(a, b) elif op_str == 'Power': return Power(a, b)
def subtract(left_node, right_node, name=None): # type: (NodeInput, NodeInput, str) -> Node """Return node which applies f(x) = A-B to the input nodes elementwise.""" return Subtract(left_node, right_node)