I am creating a Keras GRU model as in this following codes
inputs = Input(batch_shape=(args.batch_size, 1, args.train_n_items + 1)) masks = Masking(mask_v