diff options
Diffstat (limited to 'ethosu/vela/numeric_util.py')
-rw-r--r-- | ethosu/vela/numeric_util.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/numeric_util.py b/ethosu/vela/numeric_util.py index 1722adc2..9adf5ecb 100644 --- a/ethosu/vela/numeric_util.py +++ b/ethosu/vela/numeric_util.py @@ -85,3 +85,6 @@ def clamp_sigmoid(x): else: y = 1 / (1 + math.exp(-x)) return y + +def full_shape(dim, shape, fill): + return ([fill] * (dim - len(shape))) + shape
\ No newline at end of file |