golem ~master (2021-09-25T16:54:00Z)
Dub
Repo
multicastOp
golem
math
Undocumented in source.
auto
multicastOp
(Tensor!(T, Shape1, useGrad1) x, Tensor!(T, Shape2, useGrad2) y)
template
multicastOp
(string op)
version(all)
auto
multicastOp
(
T
size_t
[]
Shape1
UseGradient
useGrad1
size_t
[]
Shape2
UseGradient
useGrad2
)
(
Tensor
!(
T
,
Shape1
,
useGrad1
)
x
,
Tensor
!(
T
,
Shape2
,
useGrad2
)
y
)
if
(
Shape1
[0 ..
trimRightOneDims
(
Shape2
).
length
] ==
trimRightOneDims
(
Shape2
)
)
if
(
op
== "+" ||
op
== "-"
)
template
multicastOp
(string op)
Members
Functions
multicastOp
auto
multicastOp
(Tensor!(T, Shape1, useGrad1) x, Tensor!(T, Shape2, useGrad2) y)
Undocumented in source. Be warned that the author may not have intended to support it.
Meta
Source
See Implementation
golem
math
functions
acosh
asinh
batchSum
concat
concat2D
conv2D
conv2DShape
cosh
dropout
exp
flatten
leakyRelu
linear
log
makeConcatShape
mean
mergeEvenOdd2D
projection1D
relu
sigmoid
sinh
softmax
softmaxCrossEntropy
softplus
splitEvenOdd2D
sum
tanh
templates
ConcatTensor
broadcastOp
multicastOp