Skip to content

Commit 761a387

Browse files
authored
Merge pull request #23 from michuanhaohao/patch-1
Update layers.py --> Conv1dLayer by @michuanhaohao
2 parents 39ebf88 + b9caed8 commit 761a387

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/layers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -960,7 +960,7 @@ class Conv1dLayer(Layer):
960960
The `Layer` class feeding into this layer, [batch, in_width, in_channels].
961961
act : activation function, None for identity.
962962
shape : list of shape
963-
shape of the filters, [filter_height, filter_width, in_channels, out_channels].
963+
shape of the filters, [filter_length, in_channels, out_channels].
964964
strides : a list of ints.
965965
The stride of the sliding window for each dimension of input.\n
966966
It Must be in the same order as the dimension specified with format.
@@ -1004,7 +1004,7 @@ def __init__(
10041004
W = tf.get_variable(name='W_conv1d', shape=shape, initializer=W_init, **W_init_args )
10051005
if b_init:
10061006
b = tf.get_variable(name='b_conv1d', shape=(shape[-1]), initializer=b_init, **b_init_args )
1007-
self.outputs = act( tf.nn.conv1d(self.inputs, W, strides=strides, padding=padding,
1007+
self.outputs = act( tf.nn.conv1d(self.inputs, W, stride=strides, padding=padding,
10081008
use_cudnn_on_gpu=use_cudnn_on_gpu, data_format=data_format) + b ) #1.2
10091009
else:
10101010
self.outputs = act( tf.nn.conv1d(self.inputs, W, strides=strides, padding=padding,

0 commit comments

Comments
 (0)