polishing/shortening

This commit is contained in:
Philip Eisenlohr 2021-02-23 19:01:10 -05:00
parent 0b08b3d233
commit 2dea705fa9
1 changed files with 2 additions and 2 deletions

View File

@ -314,8 +314,8 @@ class ConfigMaterial(Config):
shaped['v'] = np.broadcast_to(1/n,(N,n))
for k,v in shaped.items():
obj = np.broadcast_to(v.reshape(util.shapeshifter(v.shape,(N,n,4),mode='right')),(N,n,4)) if k=='O' else \
np.broadcast_to(v.reshape(util.shapeshifter(v.shape,(N,n ),mode='right')),(N,n))
target = (N,n,4) if k=='O' else (N,n)
obj = np.broadcast_to(v.reshape(util.shapeshifter(v.shape,target,mode='right')),target)
for i in range(N):
if k in ['phase','O','v']:
for j in range(n):