Skip to content

Does ot functions returns in torch tensor? #809

@98srkulkarni98

Description

@98srkulkarni98

As I know this package initially uses numpy.
And I am working with torch tensor, I would like to know all the function calls like ot.unif and ot.sinkhorn do they return tensor?
is there a simple work around

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions