search

suflaj / masked-convolution

Apache License 2.0
7 stars 3 forks source link

Masked 3D transposed convolutions #2

Closed devashishlohani closed 2 years ago

devashishlohani commented 2 years ago

Hi, Thanks for this excellent work. Can you tell me how can we implement masked 3D transposed convolutions ? Thanks a lot in advance.

suflaj commented 2 years ago

This module is not concerned with those, and judging by the fact that transposed convolutions are a fair bit different than ordinary convolutions another one should probably be created.

As this is in minimal maintenance mode I doubt I would personally implement it, but contribution is welcome if someone wishes to add this functionality.

You can resubmit an issue as a feature request if you want.

EDIT: Alternatively, if all you want is to mask just the kernel, then all that is needed is adding the transposed convolution class to the list of allowed classes, in which case I suggest you also submit a feature request and I'll add them, however note that this is not the reverse operation of a masked convolution.