Skip to content

Commit fc4945b

Browse files
author
Omegastick
committed
Update to PyTorch 1.2.0
1 parent a451ad7 commit fc4945b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/distributions/normal.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ torch::Tensor Normal::sample(c10::ArrayRef<int64_t> sample_shape)
4040
{
4141
auto shape = extended_shape(sample_shape);
4242
auto no_grad_guard = torch::NoGradGuard();
43-
return torch::normal(loc.expand(shape), scale.expand(shape));
43+
return at::normal(loc.expand(shape), scale.expand(shape));
4444
}
4545

4646
TEST_CASE("Normal")

0 commit comments

Comments
 (0)