I want second option :
tensor([[[ 1.4429, -0.8365, -0.2565,  0.2319],
         [ 0.0000,  0.0000,  0.0000,  0.0000],
         [ 0.0000,  0.0000,  0.0000,  0.0000]],

        [[ 1.7247,  0.2831,  0.0687,  0.6557],
         [ 0.5110,  1.7607,  0.8107,  0.3624],
         [ 0.0000,  0.0000,  0.0000,  0.0000]]])