api
api
org.
diffkt
abs()
all
Axes
atan()
base
Primal()
broadcast
All()
broadcast
To
Shape()
Combinators
combine
Hash()
compare
To()
Comparison
Kind
EQ
GE
GT
LE
LT
NE
concat()
conv2d()
Convolve
cos()
cross
Entropy
Loss()
cross
Entropy
Loss
From
One
Hot()
Derivative
ID
Device
CPU
GPU
Differentiable
digamma()
Dim
Data
div()
DScalar
DTensor
embedding()
eq()
exp()
expand()
expand
To
Tangent()
flatten()
flip()
Float
Scalar
Float
Tensor
forward
Derivative()
forward
Derivative1()
forward
Derivative2()
forward
Derivative3()
forward
Derivative4()
forward
Diff()
gather()
ge()
gt()
if
Then
Else()
inner
Product()
integral()
Integral
Reverse
Scalar
int
Scalar
Of()
Int
Tensor
int
Tensor
Of()
inverted
is
Prefix()
jvp()
Lazy
List
le()
left
Transpose()
lgamma()
ln()
log
Softmax()
lt()
matdiv()
matmul()
max()
meld()
min()
minus()
ne()
nll
Loss
From
One
Hot()
No
Derivative
ID
On
Device
Operations
outer
Product()
plus()
polygamma()
pow()
pr()
primal()
primal
And
Forward
Derivative()
primal
And
Gradient()
primal
And
Jvp()
primal
And
Reverse
Derivative()
primal
And
Reverse
Derivative
Impl()
primal
And
Reverse
Derivative
Transposed()
primal
And
Vjp()
product()
relu()
relu
Grad()
reshape()
reverse
Derivative()
reverse
Derivative1()
reverse
Derivative2()
reverse
Derivative3()
reverse
Derivative4()
reverse
Derivative
Transposed()
reverse
Diff()
right
Transpose()
scatter()
Shape
sigmoid()
sin()
slice()
softmax()
Sparse
Sparse
Float
Tensor
Sparse
Float
Vector
Sparse
Row
Float
Tensor
split()
sqrt()
squeeze()
stack()
stats()
Strided
Float
Tensor
sum()
tan()
tanh()
Tensor
Info
tensor
Of()
times()
transpose()
unary
Minus()
unary
Plus()
unsqueeze()
view()
vjp()
with
Change()
Wrappable
Wrapper
org.
diffkt.
external
Dnnl
Gpu
Predicate
Sparse
Ops
org.
diffkt.
forward
Forward
Derivative
ID
Forward
Scalar
Forward
Tensor
org.
diffkt.
gpu
Gpu
Float
Scalar
Gpu
Float
Tensor
org.
diffkt.
model
Activation
Adam
Optimizer
Affine
Transform
avg
Pool()
Avg
Pool2d
avg
Pool
Grad()
batch
Norm()
Batch
Norm2d
BATCHNORM_EPSILON
Batch
Norm
Result
Batch
Norm
Training
Batch
Norm
Training
Base
Batch
Norm
Training
V1
batch
Norm
Train
V1()
batch
Norm
Train
V2()
Conv2d
Conv2d
With
Same
Padding
Dense
Dropout
Embedding
Embedding
Bag
Fan
In
Fan
Mode
Fan
Out
Fixed
Learning
Rate
Optimizer
Flatten
freeze
Batch
Norm()
GRU
Initializer
into()
Layer
Layer
Single
Input
Layer
With
Inference
Mode
Linear
After
Reset
Gru
Linear
Before
Reset
GRU
max
Pool()
Max
Pool2d
Model
momentum
Updated()
Optimizer
Recurrent
Base
Relu
Layer
RMSprop
Optimizer
Sequential
SGDOptimizer
Trainable
Trainable
Component
Trainable
Layer
Trainable
Layer
Single
Input
Trainable
Tensor
org.
diffkt.
random
cauchy()
chi
Square()
Diffkt
Random
gamma()
gamma
With
Rate()
gamma
With
Scale()
gaussian()
Random
Key
org.
diffkt.
reverse
Reverse
Derivative
ID
Reverse
Scalar
Reverse
Tensor
org.
diffkt.
tracing
dedag()
Dedagged
Tracing
Tensor
eval()
jit()
Jit
Evaluator
To
Use
Best
Available
Jvm
None
Normal
Scalar
Jitted
Function
printed
Form()
Printed
Tensor
raw
Printed
Form()
simplify()
topological
Sort()
Traceable
Trace
Id
Tracing
Random
Key
Tracing
Scalar
Tracing
Tensor
Tracing
Tensor
Operations
Tracing
Visitor
use
Counts()
api
/
org.diffkt
/
FloatTensor
/
zip
zip
open
fun
zip
(
right
:
FloatTensor
,
f
:
(
Float
,
Float
)
->
Float
)
:
FloatTensor
Content copied to clipboard