获取块大小 ¶

class torch.ao.quantization.observer.get_block_size(input_shape, granularity)

根据输入形状和粒度类型获取块大小。

参数:

• 输入形状（元组[int, ...]）- 输入张量的形状，可能超过 2 维

• 粒度（粒度）- 量化的粒度类型

返回类型:

元组[int, …]