get_block_size

class torch.ao.quantization.observer.get_block_size(input_shape, granularity) [исходный код]

Получить размер блока на основе формы входных данных и типа гранулярности.

Параметры:

input_shape (tuple[int, ...]) – Форма входного тензора, возможно, имеющего более двух измерений
granularity (Granularity) – Тип гранулярности квантования

Тип возвращаемого значения: