[PT FE] Fix issue with inf produced by normal distribution (openvinot…

…oolkit#21403)
xufang-lisa · Dec 1, 2023 · 7ff5174 · 7ff5174
1 parent c1cfb00
commit 7ff5174
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/frontends/pytorch/src/op/rand.cpp b/src/frontends/pytorch/src/op/rand.cpp
@@ -39,7 +39,7 @@ OutputVector make_random_normal(const NodeContext& context,
     const uint64_t seed_1 = distrib(gen);
     const uint64_t seed_2 = distrib(gen);
 
-    auto min_val = context.mark_node(v0::Constant::create(target_type, Shape{1}, {0}));
+    auto min_val = context.mark_node(v0::Constant::create(target_type, Shape{1}, {std::numeric_limits<float>::min()}));
     auto max_val = context.mark_node(v0::Constant::create(target_type, Shape{1}, {1}));
 
     auto uniform_1 = context.mark_node(