I am trying to multiply each layer of a tensor with the first layer of the tensor.
x1 = bert_model_1([x1_in, x2_in]) x1_begin = Lambda(lambda x: x[:,0])(x1) #obta