diff --git a/mindspore/nn/wrap/__init__.py b/mindspore/nn/wrap/__init__.py index 76840c5603c..09ea8fca6d7 100644 --- a/mindspore/nn/wrap/__init__.py +++ b/mindspore/nn/wrap/__init__.py @@ -21,8 +21,10 @@ from .cell_wrapper import TrainOneStepCell, WithLossCell, WithGradCell, WithEval ParameterUpdate, GetNextSingleOp, VirtualDatasetCellTriple from .loss_scale import TrainOneStepWithLossScaleCell, DynamicLossScaleUpdateCell, FixedLossScaleUpdateCell from .grad_reducer import DistributedGradReducer +from ..layer.timedistributed import TimeDistributed __all__ = [ + "TimeDistributed", "TrainOneStepCell", "WithLossCell", "WithGradCell", diff --git a/mindspore/ops/operations/other_ops.py b/mindspore/ops/operations/other_ops.py index 05140d4bd22..6b792dd6320 100644 --- a/mindspore/ops/operations/other_ops.py +++ b/mindspore/ops/operations/other_ops.py @@ -128,13 +128,13 @@ class BoundingBoxEncode(PrimitiveWithInfer): ``Ascend`` ``GPU`` Examples: - >>> anchor_box = Tensor([[4, 1, 2, 1], [2, 2, 2, 3]], mindspore.float32) - >>> groundtruth_box = Tensor([[3, 1, 2, 2], [1, 2, 1, 4]], mindspore.float32) + >>> anchor_box = Tensor([[2, 2, 2, 3], [2, 2, 2, 3]], mindspore.float32) + >>> groundtruth_box = Tensor([[1, 2, 1, 4], [1, 2, 1, 4]], mindspore.float32) >>> boundingbox_encode = ops.BoundingBoxEncode(means=(0.0, 0.0, 0.0, 0.0), stds=(1.0, 1.0, 1.0, 1.0)) >>> output = boundingbox_encode(anchor_box, groundtruth_box) >>> print(output) - [[ 5.0000000e-01 5.0000000e-01 -6.5504000e+04 6.9335938e-01] - [-1.0000000e+00 2.5000000e-01 0.0000000e+00 4.0551758e-01]] + [[ -1. 0.25 0. 0.40551758] + [ -1. 0.25 0. 0.40551758]] """ @prim_attr_register