comp-imaging / ProxImaL

A domain-specific language for image optimization.
MIT License
112 stars 29 forks source link

initializing subsample in 1d #22

Closed adler-j closed 8 years ago

adler-j commented 8 years ago

Initializing a proximal.subsample in 1d seems to have harsher requirements on the input than for example Variable:

>>> x = proximal.Variable(4)
>>> proximal.subsample(x, 2)  # this fails
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "e:\github\proximal\proximal\lin_ops\subsample.py", line 12, in __init__
    shape = tuple([(dim - 1) // step + 1 for dim, step in zip(arg.shape, steps)])
TypeError: zip argument #2 must support iteration
>>> proximal.subsample(x, [2])  # this works
SteveDiamond commented 8 years ago

This is fixed.