Fix issue: #1962, input shape problem
Hi,

To Fix #1962 
The input's shape seem to cause error in 2.2.0 version tf_albert_model

Hope that it can help.
# [Codecov](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=h1) Report
> Merging [#1966](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=desc) into [master](https://codecov.io/gh/huggingface/transformers/commit/cc7968227e08858df4a5c618c739e1a3ca050196?src=pr&el=desc) will **decrease** coverage by `0.01%`.
> The diff coverage is `100%`.

[![Impacted file tree graph](https://codecov.io/gh/huggingface/transformers/pull/1966/graphs/tree.svg?width=650&token=9qOlN6Hb1c&height=150&src=pr)](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=tree)

```diff
@@            Coverage Diff             @@
##           master    #1966      +/-   ##
==========================================
- Coverage   84.26%   84.24%   -0.02%     
==========================================
  Files         104      104              
  Lines       15431    15431              
==========================================
- Hits        13003    13000       -3     
- Misses       2428     2431       +3
```


| [Impacted Files](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=tree) | Coverage Δ | |
|---|---|---|
| [transformers/modeling\_tf\_albert.py](https://codecov.io/gh/huggingface/transformers/pull/1966/diff?src=pr&el=tree#diff-dHJhbnNmb3JtZXJzL21vZGVsaW5nX3RmX2FsYmVydC5weQ==) | `85.49% <100%> (ø)` | :arrow_up: |
| [transformers/tests/modeling\_tf\_common\_test.py](https://codecov.io/gh/huggingface/transformers/pull/1966/diff?src=pr&el=tree#diff-dHJhbnNmb3JtZXJzL3Rlc3RzL21vZGVsaW5nX3RmX2NvbW1vbl90ZXN0LnB5) | `95.63% <0%> (-1.46%)` | :arrow_down: |

------

[Continue to review full report at Codecov](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=continue).
> **Legend** - [Click here to learn more](https://docs.codecov.io/docs/codecov-delta)
> `Δ = absolute <relative> (impact)`, `ø = not affected`, `? = missing data`
> Powered by [Codecov](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=footer). Last update [cc79682...a1aec9c](https://codecov.io/gh/huggingface/transformers/pull/1966?src=pr&el=lastupdated). Read the [comment docs](https://docs.codecov.io/docs/pull-request-comments).
<|||||>Great, thank you for this @billpku !