ivy Container Example

ivy-llc · Oct 9, 2023 · 8002d7c · 8002d7c
1 parent d7cf1a5
commit 8002d7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ivy/data_classes/container/experimental/creation.py b/ivy/data_classes/container/experimental/creation.py
@@ -1283,14 +1283,14 @@ def unsorted_segment_mean(
         >>> num_segments = 1
         >>> result = x.unsorted_segment_mean(segment_ids, num_segments)
         >>> result
-        ivy.array([2.5])
+        ivy.Container([2.5])
 
         >>> y = ivy.Container([1.0, 2.0, 3.0, 4.0, 5.0, 6.0])
         >>> segment_ids = ivy.Container([0, 0, 1, 1, 2, 2])
         >>> num_segments = 3
         >>> result = y.unsorted_segment_mean(segment_ids, num_segments)
         >>> result
-        ivy.array([1.5, 3.5, 5.5])
+        ivy.Container([1.5, 3.5, 5.5])
         """
         return self.static_unsorted_segment_mean(
             self,