torch.quantized_max_pool1d¶
- torch.quantized_max_pool1d(input, kernel_size, stride=[], padding=0, dilation=1, ceil_mode=False) Tensor ¶
对由多个输入平面组成的输入量化张量应用 1D 最大池化。
- 参数
- 返回值
应用了 max_pool1d 的量化张量。
- 返回值类型
示例
>>> qx = torch.quantize_per_tensor(torch.rand(2, 2), 1.5, 3, torch.quint8) >>> torch.quantized_max_pool1d(qx, [2]) tensor([[0.0000], [1.5000]], size=(2, 1), dtype=torch.quint8, quantization_scheme=torch.per_tensor_affine, scale=1.5, zero_point=3)