deepmd.tf.model.model_stat

Alias for backward compatibility.

Module Contents

Functions

_make_all_stat_ref(data, nbatches)

make_stat_input(data, nbatches[, merge_sys])

Pack data for statistics.

merge_sys_stat(all_stat)

deepmd.tf.model.model_stat._make_all_stat_ref(data, nbatches)[source]
deepmd.tf.model.model_stat.make_stat_input(data, nbatches, merge_sys=True)[source]

Pack data for statistics.

Parameters:
data

The data

nbatchesint

The number of batches

merge_sysbool (True)

Merge system data

Returns:
all_stat:

A dictionary of list of list storing data for stat. if merge_sys == False data can be accessed by

all_stat[key][sys_idx][batch_idx][frame_idx]

else merge_sys == True can be accessed by

all_stat[key][batch_idx][frame_idx]

deepmd.tf.model.model_stat.merge_sys_stat(all_stat)[source]