Skip to content

Using JuliaFunction.apply() from Python only takes one tensor input argument #12

@ShantMahserejian

Description

@ShantMahserejian

When calling JuliaFunction.apply() from Python, and when trying to pass multiple input arguments including tensors, we are limited to passing only a single tensor variable that needs to be packed/unpacked outside/inside the Julia function call being created. For example,

JuliaFunction.apply(f, x)
where x must be a single torch tensor.

Possible solutions for lazy python implementation can include:

  1. Ideally, enable keyword arguments JuliaFunction.apply(f, *args, **kwargs) to allow multiple, labeled input arguments
  2. At least, enable non-keyword arguments JuliaFunction.apply(f, *args) to allow multiple input arguments without having to pack/unpack

This is nice to have, but not blocking.

Metadata

Metadata

Assignees

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