Skip to content

Broadcasting Autograd

gchanan edited this page Jun 15, 2017 · 19 revisions
Function Status Has fallback
copy doesn't exist in autograd N/A
apply doesn't exist in autograd N/A
apply2 doesn't exist in autograd N/A
lt done yes
lt_ doesn't exist in autograd N/A
gt done yes
gt_ doesn't exist in autograd N/A
le done yes
le_ doesn't exist in autograd N/A
ge done yes
ge_ doesn't exist in autograd N/A
eq done yes
ne done yes
ne_ doesn't exist in autograd N/A
min done yes
max done yes
masked_fill done yes
masked_scatter_
dist done yes
atan2 doesn't exist in autograd N/A
atan2_ doesn't exist in autograd N/A
pow done yes
pow_ doesn't exist in autograd N/A
lerp done yes
lerp_ doesn't exist in autograd N/A
add done yes
add_ done yes
sub done yes
sub_ done yes
mul done yes
mul_ only supports scalar multiplication N/A
div done yes
div_ only supports scalar multiplication N/A
fmod done yes
fmod_ doesn't exist in autograd N/A
remainder done yes
remainder_ doesn't exist in autograd N/A
addmm done no
addmv done no
addr done no
addbmm done no
baddbmm done no
addcmul done yes
addcmul_ done yes
addcdiv done yes
addcdiv_ done yes