这里给出的profiler用法已经比较老了，profiler提供的很多功能头用不了，最新的profiler功能可以用下面这种方法：
tf.profiler.experimental.start(log_dir) #训练开始前执行
for b in range(batch):
with tf.profiler.experimental.Trace(name='自定义名字',step_num=b):
#你的训练代码
tf.profiler.experimental.stop() #训练结束后执行，保存profile数据