[torch 参数更多]torch.distributions.transforms.ReshapeTransform

torch.distributions.transforms.ReshapeTransform

torch.distributions.transforms.ReshapeTransform(in_shape, out_shape, cache_size=0)

paddle.distribution.ReshapeTransform

paddle.distribution.ReshapeTransform(in_event_shape, out_event_shape)

PyTorch 相比 Paddle 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
in_shape in_event_shape Reshape 前的事件形状,仅参数名不一致。
out_shape out_event_shape Reshape 后的事件形状,仅参数名不一致。
cache_size - cache 大小,Paddle 无此参数,一般对网络训练结果影响不大,可直接删除。