如何在Tensorflow C ++中有效地复制行

时间:2018-05-18 23:24:24

标签: c++ tensorflow

如何在C ++ tensorflow::Tensor中多次有效地复制行?

在Eigen中,我可以执行类似M.row(1) = M.row(0)的操作,但是Tensors似乎没有.row功能。

1 个答案:

答案 0 :(得分:0)

显然,您可以使用chip进行切片和分配:

M.chip(1, 0) = M.chip(0, 0)