search

pymc-devs / pytensor

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.
https://pytensor.readthedocs.io
Other
325 stars 97 forks source link

Missing conv documentation #275

Open thipokKub opened 1 year ago

thipokKub commented 1 year ago

Issue with current documentation:

I was looking for convolutional operator, but the PyTensor's document seems to be missing

Idea or request for content:

Add how to use conv operator in the document

thipokKub commented 1 year ago

From this question. I use the pytensor.tensor.conv module and got the following error

NotImplementedError: No JAX conversion for the given `Op`: AbstractConv2d{convdim=2, border_mode=(0, (2, 1)), subsample=(1, 1), filter_flip=True, imshp=(None, None, None, None), kshp=(None, None, None, None), filter_dilation=(1, 1), num_groups=1, unshared=False}

It seems to related to another question. Jax support convolution, but the conversion from pytensor seems to be lacking. I am not sure if it is a direct mapping or not?

jessegrabowski commented 1 year ago

The original Aesara (Theano?) API docs for the tensor.nnet.conv sub-module are on the wayback machine here:

https://web.archive.org/web/20220519051518/https://aesara.readthedocs.io/en/latest/library/tensor/nnet/conv.html

Could do a copy/paste job? I know even less about docs than anything else.

ricardoV94 commented 1 year ago

They are certainly still there in the git history ;)

jessegrabowski commented 1 year ago

I thought I was so clever : (