More effcient way to generate unit vector multiples (no for loop)?

1 view (last 30 days)
Hi, I am trying to generate unit vectors of N dimensions, and all the multiples of these unit vectors out to some distance, d. I want to return both the positive and negative directions. The only way I can do this so far is with the for loop below, which gets slow for large d. Order of vectors in final matrix is unimportant. Any tips would be greatly appreciated.
tic;
d=1e5;
N=25;
unit_vecs=zeros(2*N*d,N); %pre-allocation
I=speye(N);
for i=1:d
unit_vecs(((i-1)*2*N+1):(i*2*N),:)=cat(1,i*I,-i*I);
end
toc

Accepted Answer

Andrei Bobrov
Andrei Bobrov on 7 Oct 2015
unit_vecs = kron((1:d)',[I;-I]);

More Answers (0)

Categories

Find more on Performance and Memory in Help Center and File Exchange

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!