TensorFlow:内存分配对齐

时间:2018-03-20 20:29:52

标签: python tensorflow memory-management

在TensorFlow中,当我创建一个这样的变量时,

kernel = tf.Variable(tf.truncated_normal([3, 3, 2, 1],
                     dtype=tf.float32,stddev=1e-1), name='weights')

18个32位浮点变量分配在72(3 * 3 * 2 * 1 * 4)个连续字节中。如何以这样的方式对齐这些变量:首先将3 * 3浮点值转到64字节块的前36个字节,然后将下一个3 * 3浮点值分配给下一个64字节的前36个字节块。到目前为止,我只关心CPU。

0 个答案:

没有答案