xlnstorch.csrc.matmul_forward

Contents

xlnstorch.csrc.matmul_forward#

xlnstorch.csrc.matmul_forward(A, B, base)#

Perform LNS matrix multiplication.

Parameters:
  • A (torch.Tensor) – The first tensor in LNS format (torch.int64).

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

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

Return type:

torch.Tensor