Skip to content

[doc] Tensor.mean: dtype kwarg is not documented #29758

@ssnl

Description

@ssnl

The following works:

>>> torch.__version__
'1.4.0a0+5635a72'

>>> x.mean(dtype=torch.float)
tensor(1.5000)
>>> help(x.mean)

>>> x.mean(dim=1, dtype=torch.float)
tensor([1.6667, 1.3333])

Yet the doc doesn't say anything about the dtype argument:

Help on built-in function mean:

mean(...) method of torch.Tensor instance
    mean(dim=None, keepdim=False) -> Tensor or (Tensor, Tensor)

    See :func:`torch.mean`

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: docsRelated to our documentation, both in docs/ and docblocksmodule: reductionstriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions