Merge branch 'dygraph' into pgnet-v1

2026-10-28 02:06:55 +08:00 · 2021-03-18 16:46:56 +08:00
parent 93c919e644 9749cad317
commit 1bbc3f6872
5 changed files with 19 additions and 8 deletions
@@ -38,7 +38,7 @@ class AttentionHead(nn.Layer):
        return input_ont_hot

    def forward(self, inputs, targets=None, batch_max_length=25):
-        batch_size = inputs.shape[0]
+        batch_size = paddle.shape(inputs)[0]
        num_steps = batch_max_length

        hidden = paddle.zeros((batch_size, self.hidden_size))