PyTorch torch.fake_quantize_per_tensor_affine 函数

PyTorch torch.fake_quantize_per_tensor_affine 函数 Pytorch torch 参考手册 torch.fake_quantize_per_tensor_affine 是 PyTorch 中用于对张量进行张量级假量化（Fake Quantization）的函数。函数定义 torch.fake_quantize_per_tensor_affine(input, scale, zero..

Updated 7/16/2026

Tutorial detail pages default to noindex.

PyTorch torch.fake_quantize_per_tensor_affine 函数

Pytorch torch 参考手册

torch.fake_quantize_per_tensor_affine 是 PyTorch 中用于对张量进行张量级假量化（Fake Quantization）的函数。

函数定义

torch.fake_quantize_per_tensor_affine(input, scale, zero_point, quant_min, quant_max)

使用示例

实例

import torch

# 创建输入张量
x = torch.randn(2, 3, 4, 5)

# 定义缩放因子和零点
scale = torch.tensor(1.5)
zero_point = torch.tensor(0)

# 进行张量级假量化
y = torch.fake_quantize_per_tensor_affine(x, scale, zero_point, quant_min=0, quant_max=255)
print("量化后的形状:", y.shape)

Pytorch torch 参考手册