Module: tf.nn

TensorFlow 1 version

Wrappers for primitive Neural Net (NN) Operations.

Classes

class RNNCellDeviceWrapper: Operator that ensures an RNNCell runs on a particular device.

class RNNCellDropoutWrapper: Operator adding dropout to inputs and outputs of the given cell.

class RNNCellResidualWrapper: RNNCell wrapper that ensures cell inputs are added to the outputs.

Functions

all_candidate_sampler(...): Generate the set of all classes.

atrous_conv2d(...): Atrous convolution (a.k.a. convolution with holes or dilated convolution).

atrous_conv2d_transpose(...): The transpos