diff --git a/python/paddle/tensor/linalg.py b/python/paddle/tensor/linalg.py index 2403dc766992c40729b1c150955d480bd5e99e43..c3ccb43e90f5cf0e3800a00c1216aa361f1affb2 100644 --- a/python/paddle/tensor/linalg.py +++ b/python/paddle/tensor/linalg.py @@ -440,6 +440,11 @@ def dot(x, y, name=None): """ op_type = 'dot' + # skip var type check in dygraph mode to improve efficiency + 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)