grain ~master (2021-05-15T05:53:16Z)
Dub
Repo
transform
grain
optim
CPU version of cudnn.transform
void
transform
(
T
size_t
dim
)
(
Variable
!(
T
,
dim
,
HostStorage
)
src
,
ref
Variable
!(
T
,
dim
,
HostStorage
)
dst
,
T
alpha
= 1
,
T
beta
= 0
)
Meta
Source
See Implementation
grain
optim
aliases
StateDict
functions
make
transform
update
zeroGrad
structs
AdaDelta
AdaGrad
Adam
MLP
SGD
variables
isOptimizer
CPU version of cudnn.transform