golem ~master (2021-09-25T16:54:00Z)
Dub
Repo
softmaxCrossEntropy
golem
math
Undocumented in source. Be warned that the author may not have intended to support it.
version(all)
Tensor
!(
T
, [
Shape1
[0], 1],
useGrad
)
softmaxCrossEntropy
(
T
size_t
[]
Shape1
size_t
[]
Shape2
UseGradient
useGrad
)
(
Tensor
!(
T
,
Shape1
,
useGrad
)
x
,
Tensor
!(
T
,
Shape2
,
UseGradient.no
)
y
)
if
(
Shape1.length
== 2 &&
Shape2.length
== 2
&&
Shape1
[1] ==
Shape2
[1]
)
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