在python中生成随机对称张量

时间:2018-02-21 05:22:33

标签: python random tensorflow tensor symmetric

我想生成一个相对于轴的所有排列对称的随机(高斯)张量。最后,我想要所有具有相同分布的条目,所以通过sqrt(k!)对所有排列和重新缩放进行求和的技巧,其中k是我的张量的顺序,不起作用。例如:

import numpy as np
from itertools import permutations

noise_buffer = np.random.normal(size=n*n*n).reshape(n,n,n)/np.sqrt(6);
noise = np.zeros([n,n,n]);
for i in permutations([0,1,2]):
    noise += np.transpose(noise_buffer,axes=list(i))

我可以遍历所有坐标(-1)并适时重新调整,但这很耗时。

你知道实现这个的库吗?或者你知道任何快速实施吗?

0 个答案:

没有答案