Skip to content

convolution

Module: convolution.py

This module implements convolutional layers for Neural Networks

Authors
  • Lokesh Mohanty (lokeshm@iisc.ac.in)
Version Info
  • 06/01/2025: Initial version

Conv

Bases: Conv

Performs a convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
41
42
43
44
class Conv(eqx.nn.Conv):
    """
    Performs a convolution operation
    """

Conv1d

Bases: Conv1d

Performs a 1D convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
47
48
49
50
class Conv1d(eqx.nn.Conv1d):
    """
    Performs a 1D convolution operation
    """

Conv2d

Bases: Conv2d

Performs a 2D convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
53
54
55
56
class Conv2d(eqx.nn.Conv2d):
    """
    Performs a 2D convolution operation
    """

Conv3d

Bases: Conv3d

Performs a 3D convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
59
60
61
62
class Conv3d(eqx.nn.Conv3d):
    """
    Performs a 3D convolution operation
    """

ConvTranspose

Bases: ConvTranspose

Performs a transposed convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
65
66
67
68
class ConvTranspose(eqx.nn.ConvTranspose):
    """
    Performs a transposed convolution operation
    """

ConvTranspose1d

Bases: ConvTranspose1d

Performs a 1D transposed convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
71
72
73
74
class ConvTranspose1d(eqx.nn.ConvTranspose1d):
    """
    Performs a 1D transposed convolution operation
    """

ConvTranspose2d

Bases: ConvTranspose2d

Performs a 2D transposed convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
77
78
79
80
class ConvTranspose2d(eqx.nn.ConvTranspose2d):
    """
    Performs a 2D transposed convolution operation
    """

ConvTranspose3d

Bases: ConvTranspose3d

Performs a 3D transposed convolution operation

Source code in scirex/core/dl/nn/layers/convolution.py 
83
84
85
86
class ConvTranspose3d(eqx.nn.ConvTranspose3d):
    """
    Performs a 3D transposed convolution operation
    """