Tensorflow:从矩阵中减去向量的有效方法

时间:2017-09-30 14:11:29

标签: tensorflow

我有一个MxN矩阵和一个长度为N的向量。我想要做的是从矩阵的M行中的每一行中减去该向量。这样做的显而易见的方法是改变向量,但由于所有新的内存分配,这看起来非常低效:平铺所需的时间比实际减法的时间长4倍。

有没有办法更有效地完成它,还是我必须用C ++编写自己的操作?

0 个答案:

没有答案