commit 835d8e10f33f411664cebe65d3f6a872f6cc849a
author Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Sep 04 09:46:17 2020 +0200
committer patrik.gustavsson <patrik.gustavsson@arm.com> Mon Sep 07 06:23:54 2020 +0000
tree 60c0cfd477cac7b604ade275ec82f54cbb14e9e8
parent e5cf95b8c3de4e1e4cbc7046cafd4d84c7492596

[MLBEDSW-2928] Add batching to softmax

Added batching to softmax by reshaping the input.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I0b516f9bf2410fb86372b229beba4a7280c498cc

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 83dc61a..49521e7 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -751,7 +751,7 @@
         elif d == 2:
             return [self.shape[0], 1, 1, self.shape[1]]
         else:
-            return self.shape
+            return self.shape.copy()
 
     def __str__(self):
         return "<nng.Tensor '%s' shape=%s dtype=%s>" % (self.name, self.shape, self.dtype)