Skip to content

[feature] torch.polygamma : Support Tensor for argument n #55299

@kshitij12345

Description

@kshitij12345

We should be able to specify different n for each element of the input similar to scipy.special.polygamma.

https://docs.scipy.org/doc/scipy/reference/generated/scipy.special.polygamma.html

cc @mruberry

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: specialFunctions with no exact solutions, analogous to those in scipy.specialtriagedThis 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