def getYae(Xae, reverseUtt):
assert len(Xae.shape) in [3,4], 'Invalid number of dimensions for Xae: %i (must be 3 or 4)' % len(Xae.shape)
if reverseUtt:
Yae = np.flip(Xae, 1)
if len(Xae.shape) == 4:
Yae = np.flip(Yae, 2)
else:
Yae = Xae
return Yae
评论列表
文章目录