diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py index a62524c8e677c1b0ad0b361452e54457d1ebc1b3..2a6ffc0ab66d3ac82c2da02db61e137139c0dec4 100644 --- a/python/paddle/tensor/linalg.py +++ b/python/paddle/tensor/linalg.py @@ -440,6 +440,10 @@ def dot(x, y, name=None): """ op_type = 'dot' + if in_dygraph_mode(): + op = getattr(core.ops, op_type) + return op(x, y) + assert x is not None, 'x cannot be None in {}'.format(op_type) assert y is not None, 'y cannot be None in {}'.format(op_type)