deepmd.tf.nvnmd.fit.ener

Attributes

GLOBAL_TF_FLOAT_PRECISION
nvnmd_cfg

Functions

one_layer_nvnmd(→ deepmd.tf.env.tf.Tensor)

Build one layer with continuous or quantized value.

Module Contents

deepmd.tf.nvnmd.fit.ener.GLOBAL_TF_FLOAT_PRECISION

deepmd.tf.nvnmd.fit.ener.nvnmd_cfg

deepmd.tf.nvnmd.fit.ener.one_layer_nvnmd(inputs: deepmd.tf.env.tf.Tensor, outputs_size: int, activation_fn: Callable | None = tf.nn.tanh, precision: deepmd.tf.env.tf.DType = GLOBAL_TF_FLOAT_PRECISION, stddev: float = 1.0, bavg: float = 0.0, name: str = 'linear', reuse: bool | None = None, seed: int | None = None, use_timestep: bool = False, trainable: bool = True, useBN: bool = False, uniform_seed: bool = False, initial_variables: dict | None = None, mixed_prec: dict | None = None, final_layer: bool = False) → deepmd.tf.env.tf.Tensor

Build one layer with continuous or quantized value. Its weight and bias can be initialed with random or constant value.