doc: fix fake quantize per channel doc

pytorch · Jul 25, 2023 · 6cdfda7 · 6cdfda7
1 parent ffb1b4c
commit 6cdfda7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_torch_docs.py b/torch/_torch_docs.py
@@ -11995,7 +11995,7 @@ def merge_dicts(*dicts):
 add_docstr(
     torch.fake_quantize_per_channel_affine,
     r"""
-fake_quantize_per_channel_affine(input, scale, zero_point, quant_min, quant_max) -> Tensor
+fake_quantize_per_channel_affine(input, scale, zero_point, axis, quant_min, quant_max) -> Tensor
 
 Returns a new tensor with the data in :attr:`input` fake quantized per channel using :attr:`scale`,
 :attr:`zero_point`, :attr:`quant_min` and :attr:`quant_max`, across the channel specified by :attr:`axis`.