ND4J阵列视图

时间:2018-05-16 12:04:43

标签: java nd4j

我试图查看除一列之外的现有矩阵(INDArray)。

例如:
原文:
1个| 2个| 3个
4个| 5个| 6个
我想要的:
1个| 3个
4 | 6

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

编辑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));
      }
    }

0 个答案:

没有答案