|
|
@@ -143,6 +143,7 @@ class DSSMLayer(nn.Layer):
|
|
|
|
|
|
def _process_features(self, features, embeddings):
|
|
|
embedded_features = []
|
|
|
+ paddle.static.Print(features, message="Input features shape:", print_tensor_shape=True)
|
|
|
|
|
|
# 1. 处理视频静态特征 (使用Embedding)
|
|
|
for i in range(self.feature_num):
|
|
|
@@ -180,11 +181,14 @@ class DSSMLayer(nn.Layer):
|
|
|
)
|
|
|
feature = paddle.cast(feature, dtype='float32')
|
|
|
embedded = embeddings[self.feature_num + self.stat_features_num + i](feature)
|
|
|
+ # 将Linear层输出reshape为 [-1, 1, embedding_dim]
|
|
|
+ embedded = paddle.reshape(embedded, [-1, 1, self.embedding_dim])
|
|
|
embedded_features.append(embedded)
|
|
|
|
|
|
# 将所有embedded特征连接起来
|
|
|
return paddle.concat(embedded_features, axis=1)
|
|
|
|
|
|
+
|
|
|
def forward(self, left_features, right_features, is_infer=False):
|
|
|
if is_infer:
|
|
|
left_vec = self.get_leftvectors(left_features)
|
often