@SimonW So would there be no way to in effect learn the size of bins for a certain set of data? I’m trying to think of a way to compute this using pytorch operations instead of pooling in that case but I can’t think of anything.