int_scaled_matmul

torchao.quantization.int_scaled_matmul(a: Tensor, b: Tensor, scales1: Tensor) → Tensor

执行带缩放因子的整数矩阵乘法。

参数:

• a (torch.Tensor) – 要相乘的第一个矩阵。

• b (torch.Tensor) – 要相乘的第二个矩阵。

• scales1 (torch.Tensor) – 结果行（或列）的缩放因子。

返回:

带缩放因子矩阵乘法的结果。

返回类型:

torch.Tensor

抛出:

AssertionError – 如果输入张量的维度不匹配预期的形状。