diff --git a/mindspore/ops/composite/__init__.py b/mindspore/ops/composite/__init__.py index eda1ccada7..e9c09c2c14 100644 --- a/mindspore/ops/composite/__init__.py +++ b/mindspore/ops/composite/__init__.py @@ -27,7 +27,7 @@ from .multitype_ops.add_impl import hyper_add from .multitype_ops.ones_like_impl import ones_like from .multitype_ops.zeros_like_impl import zeros_like from .random_ops import normal, laplace, uniform, gamma, poisson, multinomial -from .math_ops import count_nonzero, tensor_dot, batch_dot +from .math_ops import count_nonzero, tensor_dot, dot, batch_dot from .array_ops import repeat_elements, sequence_mask @@ -53,6 +53,7 @@ __all__ = [ 'clip_by_global_norm', 'count_nonzero', 'tensor_dot', + 'dot', 'batch_dot', 'repeat_elements', 'sequence_mask'] diff --git a/tests/st/ops/cpu/test_dot_op.py b/tests/st/ops/cpu/test_dot_op.py index cb17725bdf..a3c68889a7 100644 --- a/tests/st/ops/cpu/test_dot_op.py +++ b/tests/st/ops/cpu/test_dot_op.py @@ -27,7 +27,7 @@ context.set_context(mode=context.GRAPH_MODE, device_target="CPU") class NetDot(nn.Cell): def construct(self, x, y): - return C.math_ops.dot(x, y) + return C.dot(x, y) @pytest.mark.level0