paddlespeech.t2s.modules.geometry 模块

paddlespeech.t2s.modules.geometry.shuffle_dim(x, axis, perm=None)

根据给定的排列或随机地对输入张量沿轴进行重排。

Args:

x (Tensor):

输入张量。

axis (int):

要打乱的轴。

perm (List[int], ndarray, optional):

沿着axis-维度重新排序张量的顺序。它是[0, d)的一个排列，其中d是输入张量的axis-维度的大小。如果未提供，则使用随机排列。默认为None。

Returns:

张量：随机排列的张量，其形状与 x 相同。