小编Max*_*ard的帖子

在摆弄TensorFlow时,我注意到一个相对简单的任务(批量处理我们的3D加速度计数据并获取每个时期的总和)的性能相对较差.这是我运行的精髓,一旦我获得了(令人难以置信的漂亮!)时间轴功能:

import numpy as np
import tensorflow as tf
from tensorflow.python.client import timeline

# Some dummy functions to compute "features" from the data

def compute_features( data ):
    feature_functions = [
        lambda x: test_sum( x, axis = 0 ),
        lambda x: test_sum( x, axis = 1 ),
        lambda x: test_sum( x, axis = 2 ),
    ]
    return tf.convert_to_tensor( [ f( data ) for f in feature_functions ] )

def test_sum( data, axis = 0 ):
    t, v = data …