unbind support bool dtype (#52080)

* unbind support bool dtype

* replace np.array_equal

paddle/phi/kernels/cpu/unbind_kernel.cc

@@ -21,6 +21,7 @@ PD_REGISTER_KERNEL(unbind,
                    CPU,
                    ALL_LAYOUT,
                    phi::UnbindKernel,
+                   bool,
                    float,
                    double,
                    phi::dtype::float16,

paddle/phi/kernels/gpu/unbind_kernel.cu

@@ -21,6 +21,7 @@ PD_REGISTER_KERNEL(unbind,
                    GPU,
                    ALL_LAYOUT,
                    phi::UnbindKernel,
+                   bool,
                    float,
                    double,
                    phi::dtype::float16,

python/paddle/fluid/tests/unittests/test_unbind_op.py

@@ -201,6 +201,7 @@ class TestUnbindOp4(TestUnbindOp):
 
 class TestUnbindBF16Op(OpTest):
     def setUp(self):
+        paddle.disable_static()
         self._set_op_type()
         self.python_api = paddle.unbind
         self.dtype = self.get_dtype()
@@ -247,5 +248,13 @@ class TestUnbindAxisError(unittest.TestCase):
             self.assertRaises(ValueError, test_invalid_axis)
 
 
+class TestUnbindBool(unittest.TestCase):
+    def test_bool(self):
+        x = paddle.to_tensor([[True, True], [False, False]])
+        xs = paddle.unbind(x, axis=0)
+        self.assertEqual(len(xs), 2)
+        np.testing.assert_array_equal(xs[0].numpy(), [True, True])
+
+
 if __name__ == '__main__':
     unittest.main()

python/paddle/tensor/manipulation.py

@@ -2759,7 +2759,7 @@ def unbind(input, axis=0):
     Removes a tensor dimension, then split the input tensor into multiple sub-Tensors.
 
     Args:
-        input (Tensor): The input variable which is an N-D Tensor, data type being float16, float32, float64, int32 or int64.
+        input (Tensor): The input variable which is an N-D Tensor, data type being bool, float16, float32, float64, int32 or int64.
         axis (int32|int64, optional): A scalar with type ``int32|int64`` shape [1]. The dimension along which to unbind.
             If :math:`axis < 0`, the dimension to unbind along is :math:`rank(input) + axis`. Default is 0.
     Returns:
@@ -2808,7 +2808,7 @@ def unbind(input, axis=0):
         check_dtype(
             dtype,
             'unbind',
-            ['float16', 'float32', 'float64', 'int32', 'int64'],
+            ['bool', 'float16', 'float32', 'float64', 'int32', 'int64'],
             'unbind',
         )
         outs = [