I would like to run the following code in tensorflow 2:
A = [LAYER_OUT[:, :(t+1), :] for t in range(LAYER_OUT.shape[1])]
However, since
