|
@@ -89,6 +89,7 @@ class DSSMLayer(nn.Layer):
|
|
starts=[i],
|
|
starts=[i],
|
|
ends=[i+1]
|
|
ends=[i+1]
|
|
)
|
|
)
|
|
|
|
+ feature = paddle.cast(feature, dtype='int64')
|
|
embedded = embeddings[i](feature)
|
|
embedded = embeddings[i](feature)
|
|
embedded_features.append(embedded)
|
|
embedded_features.append(embedded)
|
|
|
|
|