get_block_size¶ class torch.ao.quantization.observer.get_block_size(input_shape, granularity)[source][source]¶ 根据输入形状和粒度类型获取块大小。 参数 input_shape (tuple[int, ...]) – 输入张量的形状，可能超过 2 个维度 granularity (Granularity) – 量化的粒度类型 返回类型 tuple[int, …]
