Unflatten¶

paddle.nn. Unflatten ( axis, shape, name=None ) [源代码] ¶

构造一个 Unflatten 类的可调用对象。更多信息请参见代码示例。将输入 Tensor 沿指定轴 axis 上的维度展成 shape 形状。

参数¶

axis (int) - 要展开维度的轴，作为 x.shape 的索引。

shape (list|tuple|Tensor) - 在指定轴上将该维度展成 shape，其中 shape 最多包含一个 -1，如果输入 shape 不包含 -1 ，则 shape 内元素累乘的结果应该等于 x.shape[axis]。数据类型为 int。如果 shape 的类型是 list 或 tuple，它的元素可以是整数或者形状为[]的 Tensor （0-D Tensor）。如果 shape 的类型是 Tensor，则是 1-D 的 Tensor。

name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回¶

无。

代码示例¶

          import paddle

x = paddle.randn(shape=[4, 6, 8])
shape = [2, 3]
axis = 1
unflatten = paddle.nn.Unflatten(axis, shape)
res = unflatten(x)
print(res.shape)
# [4, 2, 3, 8]

         