elasticai.creator.base_modules.conv1d#

Module Contents#

Classes#

MathOperations

Conv1d

API#

class elasticai.creator.base_modules.conv1d.MathOperations[source]#

Bases: elasticai.creator.base_modules.math_operations.Quantize, typing.Protocol

class elasticai.creator.base_modules.conv1d.Conv1d(operations: elasticai.creator.base_modules.conv1d.MathOperations, in_channels: int, out_channels: int, kernel_size: int | tuple[int], stride: int | tuple[int] = 1, padding: int | tuple[int] | str = 0, dilation: int | tuple[int] = 1, groups: int = 1, bias: bool = True, device: Any = None, dtype: Any = None)[source]#

Bases: torch.nn.Conv1d

forward(x: torch.Tensor) → torch.Tensor[source]#