Input length: 1
Sample input: [array([[[[-0.07495638,  0.5843668 , -0.9523151 , ..., -1.6382122 ,
           0.40688026, -1.6353588 ],
         [-0.30035278,  1.5858772 ,  2.3910701 , ..., -0.26787183,
           0.05055772, -0.01436524],
         [ 1.6757231 , -0.6814577 ,  0.4305769 , ..., -1.1716347 ,
           0.57180935, -0.9376497 ],
         ...,
         [-0.2106286 ,  0.99071   , -0.7164499 , ...,  0.6386613 ,
          -0.01419144, -0.38396034],
         [-1.5716608 , -0.68681073, -1.3773228 , ..., -0.82904655,
          -0.3060025 , -0.09916364],
         [ 1.7339691 , -0.06337839,  0.15406427, ..., -1.1738083 ,
           0.7842859 , -0.7128991 ]]]], dtype=float32)]