deepmd.tf.model.model_stat

Alias for backward compatibility.

Functions

_make_all_stat_ref(data, nbatches)
make_stat_input(data, nbatches[, merge_sys])	Pack data for statistics.
merge_sys_stat(all_stat)

Module Contents

deepmd.tf.model.model_stat._make_all_stat_ref(data, nbatches)

deepmd.tf.model.model_stat.make_stat_input(data, nbatches, merge_sys=True)

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)