Explains structure with collections/numpy/tensorflow tensors in it

explain_structure_tf.py

import tensorflow as tf

def explain_structure(obj, how_many=3):
    if hasattr(obj, 'shape'):
        repr = f"{type(obj)}, {obj.shape}"
        if type(obj) == tf.Tensor:
            repr = f"{repr} {obj.device}"
        return repr
    elif type(obj) in [type([]), type(())]:
        inner = obj
        if how_many > 0:
            inner = inner[:how_many]
        inner_explain = ",\n".join([f"[{i}]= ( {explain_structure(inner_obj)} )" for i, inner_obj in enumerate(inner)])
        return f"{type(obj)}, len({len(obj)}), {inner_explain}"
    else:
        return f"{type(obj)}"