CUQI-DTU / CUQIpy

https://cuqi-dtu.github.io/CUQIpy/
Apache License 2.0
41 stars 7 forks source link

Allow hyperparameters on geometries #461

Open nabriis opened 1 month ago

nabriis commented 1 month ago

Is your feature request related to a problem? Please describe.

A clear and concise description of what the problem is. Include code example if possible

import cuqi
dist = Gaussian(0,1, geometry=StepExpansion(n_steps=lambda k:k))

or alternatively

import cuqi
dist = Gaussian(0,1, geometry=lambda k: StepExpansion(n_steps=k))

Describe the solution you'd like

Would be nice to allow conditioning on geometries. Above should return a conditional on k