golem ~master (2021-09-25T16:54:00Z)
Dub
Repo
multicastOp.multicastOp
golem
math
multicastOp
Undocumented in source. Be warned that the author may not have intended to support it.
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
== "*"
)
template
multicastOp
(string op)
Meta
Source
See Implementation
golem
math
multicastOp
functions
multicastOp