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, …]
