Multiplication of a tensor with a tensor given by its diagonal

Description

This is a convenience function for scaling elements of a tensor with different numbers based on their position in the tensor.

Usage

diagmul.tensor(X,i=names(D),D,j=i,by=NULL)

Arguments

Details

Let

X_{i_1… i_d k_1 … k_d}

and

D_{j_1… j_d}

than the result is:

E_{i_1… i_d k_1 … k_d}=X_{i_1… i_d k_1 … k_d}D_{j_1… j_d}

Value

A tensor with the shape and dimensions as X with entries X_ik scaled by D_im, where i and k can represent multi-indices.

Author(s)

K. Gerald van den Boogaart

See Also

to.tensor

Examples

(A <- matrix(rep(1:3,each=3),nrow=3))
(b <- to.tensor(c(1,1/2,1/3)))
diagmul.tensor(as.tensor(A),2,as.tensor(c(1,1/2,1/3)),1)
diagmul.tensor(as.tensor(A),1,as.tensor(c(1,1/2,1/3)),1)
A %*% diag(b)
diag(b) %*% A

Results