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;
}