public LayerSensitivity(Layer layer) { Matrix weightMatrix = layer.getWeightMatrix(); this.sensitivityMatrix = new Matrix(weightMatrix.getRowDimension(), weightMatrix.getColumnDimension()); this.layer = layer; }