Skip to content

Poisson NLL loss in libtorch #19186

@interesaaat

Description

@interesaaat

🚀 Feature

There is no Poisson NLL loss in libtorch.

Motivation

Needed when running models with Poisson NLL loss using libtorch.

Pitch

I can work on it if someone advice me on which files to touch.
Where should the implementation go? In aten\native\Loss.cpp plus a binding in native_functions.yaml? Where should tests go?

Alternatives

Additional context

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureA request for a proper, new feature.module: cppRelated to C++ APImodule: nnRelated to torch.nntriagedThis 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