.dimshuffle
改变输入维度的顺序,返回原始变量的一个view.
输入是一个包含 ([0,1,...,ndim-1]) 和任意数目的 ('x') 的组合:
例如:
- (('x')):将标量变成 (1) 维数组
- ((0, 1)):与原始的 (2) 维数组相同
- ((1, 0)):交换 (2) 维数组的两个维度,形状从 (N imes M) 变 (M imes N)
- (('x', 0)):形状从 (N) 变成 (1 imes N)
- ((0, 'x')):形状从 (N) 变成 (N imes 1)
- ((2, 0, 1)): 形状从 (A imes B imes C) 变成 (C imes A imes B)
- ((0, 'x', 1)): 形状从 (A imes B) 变成 (A imes 1 imes B)
- ((1, 'x', 0)): 形状从 (A imes B) 变成 (B imes 1 imes A)
- ((1,)): 将第 (0) 维除去,除去的维度的大小必须为 (1)。形状从 (1 imes A) 变成 (A)