Position Wise Feed Forward

  • Dense Layers are applied along the last (512) dims