Assign torch.Generator
in APIs like torch.randn_like()
#128786
Labels
module: python frontend
For issues relating to PyTorch's Python frontend
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
馃殌 The feature, motivation and pitch
It is possible to add
generator
argument in APIs liketorch.randn_like()
?Because
torch.randn()
has argumentgenerator
and at the same time, in my opinion,torch.randn_like(A, generator=rng_torch)
should output the same result astorch.randn(A.shape, generator=rng_torch)
.Alternatives
No response
Additional context
No response
cc @albanD
The text was updated successfully, but these errors were encountered: