[PT FE] Fix issue with inf produced by normal distribution (#21403)

This commit is contained in:
Maxim Vafin
2023-12-01 01:46:53 +01:00
committed by GitHub
parent c1cfb00f81
commit 7ff517466c

View File

@@ -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(