Unflatten¶
构造一个 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-DTensor
)。如果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]