Call specification:
r(...)
orrand(...)
- random array of the specified size and tested precisionX
,Y
- aliases tor(3, 4)
andr(4, 3)
respectively
Status meaning:
- ✔️ - check passed
- ❌ - there was an error during the check
- ❔ - status is unclear (e.g. there's no rrule for the op, but an AD system may still be able to handle it)
call | invoke_ok | cpu_f64 | cpu_f32 | gpu_f64 | gpu_f32 | zygote_cpu | zygote_gpu | yota_cpu | yota_gpu | jet | jet_rrule | docs_ok |
---|---|---|---|---|---|---|---|---|---|---|---|---|
+(r(3, 4), r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
-(r(3, 4), r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
*(r(3, 4), r(4, 3)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
/(r(3, 4), r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ |
abs.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
acos.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
asin.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
atan.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
sin.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
cos.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
tan.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
exp.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
log.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
cbrt.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
clamp.(r(3, 4), 0.2, 0.4) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
maximum(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
minimum(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
sqrt(r(3, 3)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ |
sqrt.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
reduce(+, r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
reshape(r(3, 4), 4, 3) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
reverse(r(5,)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
transpose(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
batched_mul(r(3, 4, 5), r(4, 3)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ✔️ | ✔️ |
batched_transpose(r(3, 4, 5)) | ✔️ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
batched_adjoint(r(3, 4, 5)) | ✔️ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
scatter(+, [10, 100, 1000], [3, 1, 2]) | ✔️ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
gather([1, 20, 300, 4000], [2, 4, 2]) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
conv(r(10, 10, 3, 1), r(3, 3, 3, 6)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ✔️ | ✔️ |
depthwiseconv(r(10, 10, 3, 1), r(3, 3, 3, 3)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ |
maxpool(r(8, 8, 3, 1), (2, 2)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ |
meanpool(r(8, 8, 3, 1), (2, 2)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ |
softmax(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ❌ | ❌ | ✔️ |
call | invoke_ok | cpu_f64 | cpu_f32 | gpu_f64 | gpu_f32 | zygote_cpu | zygote_gpu | yota_cpu | yota_gpu | jet | jet_rrule | docs_ok |
---|---|---|---|---|---|---|---|---|---|---|---|---|
σ.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
hardσ.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
hardtanh.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
relu.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
leakyrelu.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
relu6.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
rrelu.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
elu.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
gelu.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
swish.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
selu.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
celu.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
softplus.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
softsign.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
logσ.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
logcosh.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
mish.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
tanhshrink.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
softshrink.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ❌ | ❌ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
trelu.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
lisht.(r(3, 4)) | ✔️ | ❔ | ❔ | ❔ | ❔ | ✔️ | ✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ |
tanh_fast.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |
sigmoid_fast.(r(3, 4)) | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ | ✔️ |