223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

standard broadcast_to behavior

This commit is contained in:
Martin Diehl 2020-06-20 16:35:22 +02:00
parent 262346ff5a
commit ebdb65d31f
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
python/damask/_rotation.py
View File

@ -156,18 +156,18 @@ class Rotation:
Rotation to which the misorientation is computed. Rotation to which the misorientation is computed.
""" """
return other@self.inversed() return other*self.inversed()
def broadcast_to(self,shape): def broadcast_to(self,shape):
if isinstance(shape,int): if isinstance(shape,int): shape = (shape,)
shape = (shape,) if self.shape == ():
N = np.prod(shape)//np.prod(self.shape,dtype=int) q = np.broadcast_to(self.quaternion,shape+(4,))
else:
q = np.block([np.repeat(self.quaternion[...,0:1],N).reshape(shape+(1,)), q = np.block([np.broadcast_to(self.quaternion[...,0:1],shape).reshape(shape+(1,)),
np.repeat(self.quaternion[...,1:2],N).reshape(shape+(1,)), np.broadcast_to(self.quaternion[...,1:2],shape).reshape(shape+(1,)),
np.repeat(self.quaternion[...,2:3],N).reshape(shape+(1,)), np.broadcast_to(self.quaternion[...,2:3],shape).reshape(shape+(1,)),
np.repeat(self.quaternion[...,3:4],N).reshape(shape+(1,))]) np.broadcast_to(self.quaternion[...,3:4],shape).reshape(shape+(1,))])
return self.__class__(q) return self.__class__(q)