In below tensor a[0] has 4 value which is >0.1, a[1] has 3 value which is >0.1 and a[2] has 1 value which is >0.1. The output is giving sum of (4+3+1=8). I want tensor of [4,3,1].
a=torch.FloatTensor([[[[0.1807, 0.5755],
          [0.2828, 0.4056]]],


        [[[0.4920, 0.0308],
          [0.8676, 0.2128]]],


        [[[0.0993, 0.9040],
          [0.0488, 0.0790]]]])