Skip to content

torch.ceil, torch.floor should accept a dtype argument #36309

@zasdfgbnm

Description

@zasdfgbnm

🚀 Feature

When a user is doing torch.ceil and torch.floor, it is likely that the user is asking for a long tensor instead of a floating tensor

torch.ceil(x, dtype=torch.long) would make this possible.

cc @mruberry @rgommers

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: numpyRelated to numpy support, and also numpy compatibility of our operatorsmodule: uxtriagedThis 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