We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a451ad7 commit fc4945bCopy full SHA for fc4945b
src/distributions/normal.cpp
@@ -40,7 +40,7 @@ torch::Tensor Normal::sample(c10::ArrayRef<int64_t> sample_shape)
40
{
41
auto shape = extended_shape(sample_shape);
42
auto no_grad_guard = torch::NoGradGuard();
43
- return torch::normal(loc.expand(shape), scale.expand(shape));
+ return at::normal(loc.expand(shape), scale.expand(shape));
44
}
45
46
TEST_CASE("Normal")
0 commit comments