BST error

[linWujl]

跑bst的时候训练到中途报如下错误，请问作者知道什么问题吗

tensorflow.python.framework.errors_impl.InvalidArgumentError: 4 root error(s) found.
(0) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102]
[[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]]
[[replica_1/model/bst_1/ExpandDims/_413]]
(1) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102]
[[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]]
[[Adam/concat_18/_1394]]
(2) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102]
[[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]]
[[add_4/ReadVariableOp_1/_996]]
(3) Invalid argument: Incompatible shapes: [102,1,1] vs. [1,3,102,102]
[[node model/bst/add_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:124) ]]
0 successful operations.
0 derived errors ignored. [Op:__inference_train_function_78818]

Errors may have originated from an input operation.
Input Source operations connected to node model/bst/add_2:
model/bst/ExpandDims_2 (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:119)
model/bst/truediv (defined at data/code/aftonwu/contUndSta/dev/transformer_layer.py:123)