INDArray gradientView = flattenedGradients.get(NDArrayIndex.point(0), NDArrayIndex.interval(paramOffsetSoFar, paramOffsetSoFar + nParamsThisVertex)); vertices[vertexIdx].setBackpropGradientsViewArray(gradientView);
INDArray gradientView = flattenedGradients.get(NDArrayIndex.point(0), NDArrayIndex.interval(paramOffsetSoFar, paramOffsetSoFar + nParamsThisVertex)); vertices[vertexIdx].setBackpropGradientsViewArray(gradientView);