Hi,
I think it has only two values because all the max are at either index 0 or 2 for your particular tensor outs. If you replace that with a completely random Tensor, you will (most likely) get all 4 values.