本文简要介绍python语言中 torch.fake_quantize_per_channel_affine
的用法。
用法:
torch.fake_quantize_per_channel_affine(input, scale, zero_point, quant_min, quant_max) → Tensor
每个通道张量有一个新的fake_quantized
使用
scale
、zero_point
、quant_min
和quant_max
,通过axis
指定的通道返回一个新的张量,其中的数据在input
假量化每个通道。例子:
>>> x = torch.randn(2, 2, 2) >>> x tensor([[[-0.2525, -0.0466], [ 0.3491, -0.2168]], [[-0.5906, 1.6258], [ 0.6444, -0.0542]]]) >>> scales = (torch.randn(2) + 1) * 0.05 >>> scales tensor([0.0475, 0.0486]) >>> zero_points = torch.zeros(2).to(torch.long) >>> zero_points tensor([0, 0]) >>> torch.fake_quantize_per_channel_affine(x, scales, zero_points, 1, 0, 255) tensor([[[0.0000, 0.0000], [0.3405, 0.0000]], [[0.0000, 1.6134], [0.6323, 0.0000]]])
参数:
返回:
返回类型:
相关用法
- Python PyTorch fake_quantize_per_tensor_affine用法及代码示例
- Python PyTorch fasterrcnn_mobilenet_v3_large_320_fpn用法及代码示例
- Python PyTorch fasterrcnn_resnet50_fpn用法及代码示例
- Python PyTorch fasterrcnn_mobilenet_v3_large_fpn用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch fft2用法及代码示例
- Python PyTorch fftn用法及代码示例
- Python PyTorch flip用法及代码示例
- Python PyTorch frombuffer用法及代码示例
- Python PyTorch float_power用法及代码示例
- Python PyTorch floor_divide用法及代码示例
- Python PyTorch fractional_max_pool3d用法及代码示例
- Python PyTorch frac用法及代码示例
- Python PyTorch freeze用法及代码示例
- Python PyTorch fp16_compress_hook用法及代码示例
- Python PyTorch fftshift用法及代码示例
- Python PyTorch flipud用法及代码示例
- Python PyTorch fliplr用法及代码示例
- Python PyTorch fp16_compress_wrapper用法及代码示例
- Python PyTorch fractional_max_pool2d用法及代码示例
- Python PyTorch fftfreq用法及代码示例
- Python PyTorch from_numpy用法及代码示例
- Python PyTorch filter_wikipedia_xml用法及代码示例
- Python PyTorch fuse_modules用法及代码示例
- Python PyTorch fmax用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.fake_quantize_per_channel_affine。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。