Skip to content

Commit 4fcd92e

Browse files
committed
[layers] fix docs of Conv1dLayer, strides is int
1 parent 56db166 commit 4fcd92e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tensorlayer/layers.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -971,9 +971,8 @@ class Conv1dLayer(Layer):
971971
act : activation function, None for identity.
972972
shape : list of shape
973973
shape of the filters, [filter_length, in_channels, out_channels].
974-
strides : a list of ints.
975-
The stride of the sliding window for each dimension of input.\n
976-
It Must be in the same order as the dimension specified with format.
974+
strides : a int.
975+
The number of entries by which the filter is moved right at each step.
977976
padding : a string from: "SAME", "VALID".
978977
The type of padding algorithm to use.
979978
use_cudnn_on_gpu : An optional bool. Defaults to True.
@@ -994,7 +993,7 @@ def __init__(
994993
layer = None,
995994
act = tf.identity,
996995
shape = [5, 5, 1],
997-
strides=[1, 1, 1],
996+
strides= 1,
998997
padding='SAME',
999998
use_cudnn_on_gpu=None,
1000999
data_format=None,

0 commit comments

Comments
 (0)