Skip to content

Commit

Permalink
doc: fix fake quantize per channel doc (#105955)
Browse files Browse the repository at this point in the history
  • Loading branch information
OrenLeung authored and pytorchmergebot committed Jul 26, 2023
1 parent 60ad46f commit f725e63
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/_torch_docs.py
Original file line number Diff line number Diff line change
Expand Up @@ -11996,7 +11996,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`.
Expand Down

0 comments on commit f725e63

Please sign in to comment.