tensor_util.py 21.2 KB