diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-09-04 09:46:17 +0200 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-09-07 06:23:54 +0000 |
commit | 835d8e10f33f411664cebe65d3f6a872f6cc849a (patch) | |
tree | 60c0cfd477cac7b604ade275ec82f54cbb14e9e8 /ethosu/vela/tensor.py | |
parent | e5cf95b8c3de4e1e4cbc7046cafd4d84c7492596 (diff) | |
download | ethos-u-vela-835d8e10f33f411664cebe65d3f6a872f6cc849a.tar.gz |
[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
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 83dc61a3..49521e7a 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -751,7 +751,7 @@ class Tensor: 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) |