Summary:
Kindly let me know if its okay and if any places i need to make a fix. Closes #18534
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18604
Differential Revision:
D14680712
Pulled By: soumith
fbshipit-source-id:
030e4a3d8f7839cbe2b8a3ef386323f0d39eb81a
end (float): the ending value for the set of points
step (float): the gap between each pair of adjacent points. Default: ``1``.
{out}
- {dtype}
+ {dtype} If `dtype` is not given, infer the data type from the other input
+ arguments. If any of `start`, `end`, or `stop` are floating-point, the
+ `dtype` is inferred to be the default dtype, see
+ :meth:`~torch.get_default_dtype`. Otherwise, the `dtype` is inferred to
+ be `torch.int64`.
{layout}
{device}
{requires_grad}