Skip to content

Possible issue with batched G_ tensor in QPFunctionFn_infeas.forward #406

@oomcth

Description

@oomcth

In bindings/python/proxsuite/torch/qplayer.py:

class QPFunctionFn_infeas(Function):
    @staticmethod
    def forward(ctx, Q_, p_, A_, b_, G_, l_, u_):
         n_in, nz = G_.size() # <- can't G be batched? If so, this will cause an issue.

If G_ can be batched (e.g. shape (batch_size, n_in, nz)), this line will break. Might need a shape check or to handle batching explicitly.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions