Skip to content

Commit 5ed9d97

Browse files
committed
Allows scalars as Param's shape.
1 parent 582dba0 commit 5ed9d97

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

DeepFried2/Param.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ class Param(object):
66

77
def __init__(self, shape, init, fan=None, name=None, learn=True, decay=True, dtype=df.floatX, **kw):
88
self.init = init
9-
self.shape = shape
9+
self.shape = (shape,) if _np.isscalar(shape) else tuple(shape)
1010
self.fan = fan
1111
self.decay = decay
1212

0 commit comments

Comments
 (0)