grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
convolution
grain
chain
tensor convolution (do cross correlation default)
auto
convolution
(
bool
isConv
= false
bool
isNchw
= true
T
size_t
dim
size_t
imDim
alias
Storage
)
(
Variable
!(
T
,
dim
,
Storage
)
x
,
Variable
!(
T
,
dim
,
Storage
)
w
,
int
[
imDim
]
stride
,
int
[
imDim
]
pad
,
int
[
imDim
]
dilation
)
Meta
Source
See Implementation
grain
chain
functions
abs
addVec
convolution
cos
crossEntropy
dropout
exp
log
logSoftmax
matMul
neg
negativeLogLikelihood
opBinaryFunc
pow
reciprocal
relu
sigmoid
sin
squeeze
sum
tan
tanh
unsqueeze
view
structs
Convolution
Embedding
Linear
tensor convolution (do cross correlation default)