z = encoder(x)
ze_bool = (quantizer(z) >= 0).long()
output = (
ze_bool.permute(0, 2, 3, 1).reshape(
ze_bool.size(0), -1, ze_bool.size(1)
)
z = encoder(x)
ze_bool = (quantizer(z) >= 0).long()
output = (
ze_bool.permute(0, 2, 3, 1).reshape(
ze_bool.size(0), -1, ze_bool.size(1)
)