Как получить гистограмму тензоров PyTorch в пакетном режиме? ⇐ Python

[Anonymous]

Есть ли способ получить гистограммы тензоров факелов в пакетном режиме?
Например:
x — тензор формы (64, 224, 224)
# x will have shape of (64, 256)
x = batch_histogram(x, bins=256, min=0, max=255)


Подробнее здесь: https://stackoverflow.com/questions/694 ... in-batches