From c189eba3e139aca43974e63dba9fb972e88c9d74 Mon Sep 17 00:00:00 2001 From: Arunava Date: Fri, 29 Mar 2019 06:36:40 -0700 Subject: [PATCH] Fixed torch.arange docs (#18604) 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 --- torch/_torch_docs.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py index 1a9262b..580481f 100644 --- a/torch/_torch_docs.py +++ b/torch/_torch_docs.py @@ -4031,7 +4031,11 @@ Args: 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} -- 2.7.4