xlnstorch.csrc.conv1d_forward

Contents

xlnstorch.csrc.conv1d_forward#

xlnstorch.csrc.conv1d_forward(x, weight, bias, base, stride, padding, dilation, groups)#

Perform LNS 1D convolution.

Parameters:

  • x (torch.Tensor) – The input tensor in LNS format (torch.int64).

  • weight (torch.Tensor) – The convolution kernel in LNS format (torch.int64).

  • bias (torch.Tensor) – The bias tensor in LNS format (torch.int64).

  • base (torch.Tensor) – The base for the LNS representation.

  • stride (int) – The stride of the convolution.

  • padding (int) – The padding applied to the input tensor.

  • dilation (int) – The dilation factor for the convolution.

  • groups (int) – The number of groups for the convolution.

Returns:

The output tensor in LNS format (torch.int64) after convolution.

Return type:

torch.Tensor

Contents