kernel_size parameter can be an int for 1d Conv and ConvTranspose #551
copybara-service
merged
5 commits into
google:master
from
freddyaboulton:547-kernel-size-int
avital
commented
on 2020-10-23
avital
commented
on 2020-10-23
jheek
requested changes
on 2020-10-23
freddyaboulton
changed the title
levskaya
approved these changes
on 2020-10-31
kernel_size can be an int for Conv and ConvTranspose.
e9dd06c3
Removing extraneous comma
b4696cfc
Mapping int to (int,) in Conv and ConvTranspose for kernel_size param.
188b1346
Updating docstring for Conv explaining that int can be passed in.
a4ac2d0c
Modifying linen Conv/Conv Transpose to accept int for kernel size and…
c5dddd40
Login to write a write a comment.
Login via GitHub