add weigth data to unit test

94952111 · root · chajchaj · e1ea895f · 94952111
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py ...n/paddle/fluid/tests/unittests/test_cross_entropy_loss.py +4 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
+++ b/python/paddle/fluid/tests/unittests/test_cross_entropy_loss.py
@@ -1373,8 +1373,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                label_data = paddle.randint(
                    0, 100, shape=[20, 1], dtype="int64")
                label_data[0] = 255
+                weight_data = paddle.rand([100])
                paddle.nn.functional.cross_entropy(
-                    input=input_data, label=label_data)
+                    input=input_data, label=label_data, weight=weight_data)

            self.assertRaises(ValueError, test_LabelValue)

@@ -1383,8 +1384,9 @@ class TestCrossEntropyFAPIError(unittest.TestCase):
                label_data = paddle.randint(
                    0, 100, shape=[20, 1], dtype="int64")
                label_data[0] = -1
+                weight_data = paddle.rand([100])
                paddle.nn.functional.cross_entropy(
-                    input=input_data, label=label_data)
+                    input=input_data, label=label_data, weight=weight_data)

            self.assertRaises(ValueError, test_LabelValueNeg)