Attention Layer In Neural Network Machine