Matrix operation
Matthew Knepley
knepley at mcs.anl.gov
Mon Feb 20 17:15:34 CST 2006
Hong Zhang <hzhang at mcs.anl.gov> writes:
I would recommend the later
MatGetDiagonal(A, diag);
VecPointwiseMult(w, diag, u);
MatMult(A, u, v);
VecAXPY(w, -1.0, v);
Matt
> Billy,
>
> You can do
> MatGetDiagonal(A,diag);
> MatMult(A,u,&vtmp);
> VecGetArray(diag,diag_array);
> VecGetArray(vtmp,vtmp_array);
> VecGetArray(u,u_arrar);
> VecGetArray(h,h_array);
> VecGetLocalSize(h,&n);
> for (i=0; i<n; i++){
> h_array[i] += diag_array[i]*u_array[i] - vtmp_array[i];
> }
> VecRestoreArray(h,h_array);
> VecRestoreArray()
> ...
>
> or you can use VecPointwiseMult() got get
> Diag(A)*u.
>
> Hong
>
> On Mon, 20 Feb 2006 billy at dem.uminho.pt wrote:
>
>>
>> Hello,
>>
>> I would like to do the following matrix-vector operation in PETSc:
>>
>> h += (Diag(A) - A) * u
>>
>> where A is a matrix, Diag(A) is a matrix and h, u are vectors. It is the
>> multiplication of a matrix with diagonals set to zero with a vector.
>>
>>
>> How can I do this in PETSc? because MatGetDiagonal gives a vector not a matrix.
>>
>>
>> Billy.
>>
>>
>>
>
>
>
--
"Failure has a thousand explanations. Success doesn't need one" -- Sir Alec Guiness
More information about the petsc-users
mailing list