0

我正在尝试查看不包括一列的现有矩阵(INDArray)。

例如:
原始:
1|2|3
4|5|6
我想要什么:
1|3
4|6

文档显示了如何获取列/行的间隔,但没有关于排除任意行/列。

编辑 1(我的临时解决方案):

INDArray matrix_ = Nd4j.create(matrix.columns(), matrix.rows() - 1);
    int j = 0;
    for (int i = 0; i < matrix.rows(); i++) {
      if (i != rowToDel){
          matrix_.putRow(j++, matrix.getRow(i));
      }
    }
4

0 回答 0