golem ~master (2021-09-25T16:54:00Z)
Dub
Repo
broadcastOp.broadcastOp
golem
math
broadcastOp
Undocumented in source. Be warned that the author may not have intended to support it.
auto
broadcastOp
(Tensor!(T, Shape1, useGrad1) x, Tensor!(T, Shape2, useGrad2) y)
template
broadcastOp
(string op)
version(all)
auto
broadcastOp
(
T
size_t
[]
Shape1
UseGradient
useGrad1
size_t
[]
Shape2
UseGradient
useGrad2
)
(
Tensor
!(
T
,
Shape1
,
useGrad1
)
x
,
Tensor
!(
T
,
Shape2
,
useGrad2
)
y
)
if
(
Shape1
[$ -
Shape2.length
.. $] ==
Shape2
)
if
(
op
== "*"
)
template
broadcastOp
(string op)
Meta
Source
See Implementation
golem
math
broadcastOp
functions
broadcastOp