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

2 次查看（过去 30 天）
Colton
Colton 2015-10-7
编辑： Colton 2015-10-8
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

请先登录，再回答此问题。

采纳的回答

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

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Performance and Memory 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by