Skip to content

[feature request] Out-variant and dtype argument for torch.argmax / torch.argmin / torch.argsort (and friends) #32570

New issue
New issue
Open
Open
[feature request] Out-variant and dtype argument for torch.argmax / torch.argmin / torch.argsort (and friends)#32570
Labels
function requestA request for a new function or the addition of new arguments/modes to an existing function.module: memory usagePyTorch is using more memory than it should, or it is leaking memorymodule: reductionsmodule: sorting and selectiontriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
@vadimkantorov

Description

@vadimkantorov
vadimkantorov
opened on Jan 24, 2020

It could save a lot of memory when relevant axis is very thin (e.g. 2 or 3 element-sized axis), and the result is known to fit in a byte (x8 memory save)

cc @heitorschueroff

Metadata

Metadata

Assignees

No one assigned

    Labels

    function requestA request for a new function or the addition of new arguments/modes to an existing function.module: memory usagePyTorch is using more memory than it should, or it is leaking memorymodule: reductionsmodule: sorting and selectiontriagedThis 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
      pFad - Phonifier reborn

      Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

      Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


      Alternative Proxies:

      Alternative Proxy

      pFad Proxy

      pFad v3 Proxy

      pFad v4 Proxy