Update monai/transforms/post/array.py

Co-authored-by: YunLiu <[email protected]>
Signed-off-by: Matthias Hadlich <[email protected]>

monai/transforms/post/array.py

@@ -960,7 +960,7 @@ def __call__(self, img: NdarrayOrTensor) -> NdarrayOrTensor:
         """
         Args:
             img: Input image on which the distance transform shall be run.
-                For CuPy shape must be ([channel, ]spatial_dim1, spatial_dim2 [, spatial_dim3]), so 2D, 3D and 4D input only.
+                channel first array, must have shape: (num_channels, H[, W, ..., ])
                 If you need to run the transform on other shapes, use the ``force_scipy`` flag.
                 4D input gets passed channel-wise to cupy.