torchlayers icon indicating copy to clipboard operation
torchlayers copied to clipboard

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks

Results 4 torchlayers issues
Sort by recently updated
recently updated
newest added

Just wondering if still maintained...

Hey! I have the following issue. Using Python 3.9.7 and torchlayers 0.1.1. Trying to do this: ``` import torchlayers as tl # Regularize all parameters of Linear module regularized_layer =...

Hi, I have a relatively straightforward situation where I need to validate my input shapes while `jit.load`ing a saved model and I can't seem to find the solution for. ```python...

is there any possibility for the 'same' padding convolution to work on even values of kernel size, stride, and dilation?

bug
enhancement