use randn

pytorch · Nov 28, 2023 · e401ad4 · e401ad4
1 parent a09ecee
commit e401ad4
Showing 1 changed file with 1 addition and 19 deletions.
diff --git a/test/test_core_aten_ops.py b/test/test_core_aten_ops.py
@@ -3296,25 +3296,7 @@ def test_aten_pow_Tensor_Scalar_2(self):
     run_export_and_compare(self, torch.ops.aten.pow.Tensor_Scalar, args, kwargs)
 
   def test_aten_pow_Scalar_1(self):
-    args = (10000,
-            torch.Tensor([
-                0.0000, 0.0000, 0.0156, 0.0156, 0.0312, 0.0312, 0.0469, 0.0469,
-                0.0625, 0.0625, 0.0781, 0.0781, 0.0938, 0.0938, 0.1094, 0.1094,
-                0.1250, 0.1250, 0.1406, 0.1406, 0.1562, 0.1562, 0.1719, 0.1719,
-                0.1875, 0.1875, 0.2031, 0.2031, 0.2188, 0.2188, 0.2344, 0.2344,
-                0.2500, 0.2500, 0.2656, 0.2656, 0.2812, 0.2812, 0.2969, 0.2969,
-                0.3125, 0.3125, 0.3281, 0.3281, 0.3438, 0.3438, 0.3594, 0.3594,
-                0.3750, 0.3750, 0.3906, 0.3906, 0.4062, 0.4062, 0.4219, 0.4219,
-                0.4375, 0.4375, 0.4531, 0.4531, 0.4688, 0.4688, 0.4844, 0.4844,
-                0.5000, 0.5000, 0.5156, 0.5156, 0.5312, 0.5312, 0.5469, 0.5469,
-                0.5625, 0.5625, 0.5781, 0.5781, 0.5938, 0.5938, 0.6094, 0.6094,
-                0.6250, 0.6250, 0.6406, 0.6406, 0.6562, 0.6562, 0.6719, 0.6719,
-                0.6875, 0.6875, 0.7031, 0.7031, 0.7188, 0.7188, 0.7344, 0.7344,
-                0.7500, 0.7500, 0.7656, 0.7656, 0.7812, 0.7812, 0.7969, 0.7969,
-                0.8125, 0.8125, 0.8281, 0.8281, 0.8438, 0.8438, 0.8594, 0.8594,
-                0.8750, 0.8750, 0.8906, 0.8906, 0.9062, 0.9062, 0.9219, 0.9219,
-                0.9375, 0.9375, 0.9531, 0.9531, 0.9688, 0.9688, 0.9844, 0.9844
-            ]))
+    args = (10000, torch.randn(16 * 8))
     kwargs = dict()
     run_export_and_compare(self, torch.ops.aten.pow.Scalar, args, kwargs)