标签: tensorflow
我有一个MxN矩阵和一个长度为N的向量。我想要做的是从矩阵的M行中的每一行中减去该向量。这样做的显而易见的方法是改变向量,但由于所有新的内存分配,这看起来非常低效:平铺所需的时间比实际减法的时间长4倍。
有没有办法更有效地完成它,还是我必须用C ++编写自己的操作?