basic_ops

Basic primitives for neural networks.

Classes

CastToDType	Casts an input to a given dtype.
Elementwise	A layer that runs an elementwise operation on its NamedArray argument.
Softmax	Layer that applies a softmax along a given set of axes.