forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 1
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 |