public boolean computeDimension(MatrixCharacteristics in, MatrixCharacteristics out) throws DMLRuntimeException { if (in.getCols() == 1) // diagV2M out.set(in.getRows(), in.getRows(), in.getRowsPerBlock(), in.getRowsPerBlock()); else // diagM2V out.set(in.getRows(), 1, in.getRowsPerBlock(), in.getRowsPerBlock()); return false; }
public boolean computeDimension(MatrixCharacteristics in, MatrixCharacteristics out) throws DMLRuntimeException { out.set(1, 1, in.getRowsPerBlock(), in.getColsPerBlock()); return true; }