deepmd.tf.utils.compress

Attributes

log

Functions

get_type_embedding(→ numpy.ndarray)
get_two_side_type_embedding(→ numpy.ndarray)
get_extra_side_embedding_net_variable(→ dict[str, Any])
_layer_0(→ deepmd.tf.env.tf.Tensor)
_layer_1(→ tuple[deepmd.tf.env.tf.Tensor, ...)
make_data(→ numpy.ndarray)

Module Contents

deepmd.tf.utils.compress.log

deepmd.tf.utils.compress.get_type_embedding(self: Any, graph: deepmd.tf.env.tf.Graph) → numpy.ndarray

deepmd.tf.utils.compress.get_two_side_type_embedding(self: Any, graph: deepmd.tf.env.tf.Graph, suffix: str = '') → numpy.ndarray

deepmd.tf.utils.compress.get_extra_side_embedding_net_variable(self: Any, graph_def: deepmd.tf.env.tf.GraphDef, type_side_suffix: str, varialbe_name: str, suffix: str) → dict[str, Any]

deepmd.tf.utils.compress._layer_0(self: Any, x: deepmd.tf.env.tf.Tensor, w: deepmd.tf.env.tf.Tensor, b: deepmd.tf.env.tf.Tensor) → deepmd.tf.env.tf.Tensor

deepmd.tf.utils.compress._layer_1(self: Any, x: deepmd.tf.env.tf.Tensor, w: deepmd.tf.env.tf.Tensor, b: deepmd.tf.env.tf.Tensor) → tuple[deepmd.tf.env.tf.Tensor, deepmd.tf.env.tf.Tensor]

deepmd.tf.utils.compress.make_data(self: Any, xx: numpy.ndarray) → numpy.ndarray