MCPcopy Index your code
hub / github.com/google-deepmind/graph_nets / _unstack

Function _unstack

graph_nets/utils_np.py:239–242  ·  view source on GitHub ↗

Similar to `tf.unstack`.

(array)

Source from the content-addressed store, hash-verified

237
238
239def _unstack(array):
240 """Similar to `tf.unstack`."""
241 num_splits = int(array.shape[0])
242 return [np.squeeze(x, 0) for x in np.split(array, num_splits, axis=0)]
243
244
245def data_dict_to_networkx(data_dict):

Callers 2

data_dict_to_networkxFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected