neograd package

Subpackages

• neograd.autograd package
  • Subpackages
    • neograd.autograd.ops package
      • Submodules
      • neograd.autograd.ops.basics module
      • neograd.autograd.ops.conv module
      • neograd.autograd.ops.operation module
      • Module contents
  • Submodules
  • neograd.autograd.graph module
    • Graph
      • Graph.add_edge()
      • Graph.add_node()
      • Graph.add_tensor()
      • Graph.get_node()
      • Graph.graph
      • Graph.remove_tensor()
      • Graph.reset_graph()
      • Graph.reset_visited()
      • Graph.zero_grad()
  • neograd.autograd.node module
    • Node
      • Node.add_child()
      • Node.add_parent()
      • Node.are_children_visited()
      • Node.are_parents_visited()
      • Node.backward()
      • Node.top_sort()
      • Node.visit_all_children()
  • neograd.autograd.tensor module
    • Tensor
      • Tensor.T
      • Tensor._backward()
      • Tensor.accumulate_grad()
      • Tensor.backward()
      • Tensor.data
      • Tensor.dot()
      • Tensor.exp()
      • Tensor.flatten()
      • Tensor.reshape()
      • Tensor.set_grad_fn()
      • Tensor.shape
      • Tensor.sum()
      • Tensor.zero_grad()
  • neograd.autograd.utils module
    • _evaluate_grad_check()
    • _wiggle_params()
    • fn_grad_check()
    • get_graph()
    • grad_check()
    • new_graph
    • no_track
    • process_data()
    • unbroadcast_data()
  • Module contents
• neograd.nn package
  • Submodules
  • neograd.nn.activations module
    • LeakyReLU
      • LeakyReLU.backward()
      • LeakyReLU.forward()
    • ReLU
      • ReLU.backward()
      • ReLU.forward()
    • Sigmoid
      • Sigmoid.backward()
      • Sigmoid.forward()
    • Softmax
      • Softmax.backward()
      • Softmax.calc_softmax()
      • Softmax.forward()
    • Tanh
      • Tanh.backward()
      • Tanh.forward()
  • neograd.nn.checkpoint module
    • Checkpoint
      • Checkpoint._generate_hash()
      • Checkpoint._init_files()
      • Checkpoint._save()
      • Checkpoint._update()
      • Checkpoint.add()
      • Checkpoint.load()
      • Checkpoint.new_session()
      • Checkpoint.specify_session()
  • neograd.nn.layers module
    • Container
      • Container.freeze()
      • Container.parameters()
      • Container.set_eval()
      • Container.set_params()
      • Container.unfreeze()
    • Layer
      • Layer.freeze()
      • Layer.parameters()
      • Layer.set_eval()
      • Layer.set_params()
      • Layer.unfreeze()
    • Param
      • Param.freeze()
      • Param.unfreeze()
  • neograd.nn.loss module
    • BCE
      • BCE.forward()
    • CE
      • CE.forward()
    • Loss
      • Loss.get_num_examples()
    • MSE
      • MSE.forward()
    • SoftmaxCE
      • SoftmaxCE.backward()
      • SoftmaxCE.forward()
  • neograd.nn.model module
    • EvalMode
    • Model
      • Model.eval()
      • Model.get_layers()
      • Model.load()
      • Model.parameters()
      • Model.save()
      • Model.set_eval()
  • neograd.nn.optim module
    • Adam
      • Adam.init_adam_grads()
      • Adam.reset_iter()
      • Adam.step()
      • Adam.update_adam_grads()
    • GD
      • GD.step()
    • Momentum
      • Momentum.init_momentum_grads()
      • Momentum.step()
      • Momentum.update_momentum_grads()
    • Optimizer
      • Optimizer.zero_grad()
    • RMSProp
      • RMSProp.init_rms_grads()
      • RMSProp.step()
      • RMSProp.update_rms_grads()
  • neograd.nn.utils module
    • get_batches()
    • load_model()
    • save_model()
  • Module contents

Module contents

neograd._NG_GRAPH = Graph()

_NG_GRAPH is the global graph object used to construct backprop graphs