basic_ops#
Basic primitives for neural networks.
Classes
Casts an input to a given dtype. |
|
A layer that runs an elementwise operation on its |
|
Layer that applies a softmax along a given set of axes. |
|
Softly rescales a value to lie within a range using tanh. |