Exports
-
network, sigmoid, SigmoidActivation, ReluActivation, relu, TanhActivation, tanh, outShape, Linear, init, inShape, LinearGate, forward, linear, outShape, Flatten, forward, inShape, init, outShape, Conv2DGate, Conv2D, init, forward, inShape, conv2d, outShape, maxpool2d, MaxPool2DGate, init, MaxPool2D, forward, inShape, gru, GRULayer, forward, GRUGate, init, outShape, EmbeddingGate, init, embedding, Embedding, forward, inShape, outShape, GCNLayer, init, gcn, GCNGate, forward, inShape, softmax_cross_entropy, SparseSoftmaxCrossEntropyLoss, sigmoid_cross_entropy, SoftmaxCrossEntropyLoss, SigmoidCrossEntropyLoss, sparse_softmax_cross_entropy, mse_loss, MSELoss, newSGD, SGD, optimizerSGDMomentum, zeroGrads, optimizerSGD, update, optimizer, update, optimizerAdam, update, Adam, optimizer, optimizer, Optimizer, SGDMomentum, yann_normal, xavier_normal, yann_uniform, kaiming_normal, xavier_uniform, kaiming_uniform