标签： python tensorflow tensorflow2.0
我正在使用具有8个内核的本地机器上的tensorflow 2.0进行一些稀疏矩阵计算。 执行该操作时，我可以在top命令中看到仅使用了8个内核中的1个。
运算是稀疏乘法
tf.sparse.sparse_dense_matmul(a_tf_sp, b_tf_dense_T)
我如何告诉tensorflow 2.0使用所有核心进行此计算。
答案 0 :(得分：0)
您可以在with tf.device('/cpu:0'):内定义代码并检查吗？
with tf.device('/cpu:0'):