fix colotensor.type() raise NotImplementedError
diff --git a/colossalai/nn/_ops/element_wise.py b/colossalai/nn/_ops/element_wise.py
index c3c1421e7fdc..462670e7278a 100644
--- a/colossalai/nn/_ops/element_wise.py
+++ b/colossalai/nn/_ops/element_wise.py
@@ -18,6 +18,8 @@ def elementwise_op(input_tensor: GeneralTensor, *args, **kwargs):
 
         output = op(input_tensor, *args, **kwargs)
         if isinstance(input_tensor, ColoTensor):
+            if isinstance(output, str):
+                return output
             if not isinstance(output, torch.Tensor):
                 raise NotImplementedError
             return ColoTensor.from_torch_tensor(output,

hi, dev from biomap. Encountered this error with calling colotensor.type()