快捷方式

quantize_affine_floatx

torchao.quantization.quantize_affine_floatx(tensor: Tensor, scale: Tensor, ebits: int, mbits: int) Tensor[source]

将 float32 高精度浮点张量量化为低精度浮点数,并将结果转换为未压缩的浮点格式,格式为 00SEEEMM(对于 fp6_e3m2),其中 S 表示符号位,e 表示指数位,m 表示尾数位

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得您的问题解答

查看资源