Skip to content

Commit 7c9fd2f

Browse files
authored
Update docs (#7104)
* update docs * update docs * update docs * update docs
1 parent 17d0d41 commit 7c9fd2f

18 files changed

+330
-25
lines changed
Lines changed: 23 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
## [ torch 参数更多 ]torch.nn.modules.module.register_module_forward_hook
1+
## [ 组合替代实现 ]torch.nn.modules.module.register_module_forward_hook
22
### [torch.nn.modules.module.register_module_forward_hook](https://pytorch.org/docs/stable/generated/torch.nn.modules.module.register_module_forward_hook.html)
33

44
```python
5-
torch.nn.modules.module.register_module_forward_hook(hook, *, prepend=False, with_kwargs=False, always_call=False)
5+
torch.nn.modules.module.register_module_forward_hook(hook, *, always_call=False)
66
```
77

88
### [paddle.nn.Layer.register_forward_post_hook](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/nn/Layer_cn.html#register-forward-post-hook-hook)
@@ -11,12 +11,28 @@ torch.nn.modules.module.register_module_forward_hook(hook, *, prepend=False, wit
1111
paddle.nn.Layer.register_forward_post_hook(hook)
1212
```
1313

14-
PyTorch 相比 Paddle 支持更多其他参数,具体如下:
14+
其中,PyTorch 为给全局所有 module 注册 hook,而 Paddle 为给单个 Layer 注册 hook。PyTorch 相比 Paddle 支持更多其他参数,具体如下:
1515
### 参数映射
1616

1717
| PyTorch | PaddlePaddle | 备注 |
1818
| ------------- | ------------ | ------------------------------------------------------ |
19-
| hook | hook | 被注册为 forward pre-hook 的函数。 |
20-
| prepend | - | 钩子执行顺序控制,Paddle 无此参数,暂无转写方式。 |
21-
| with_kwargs | - | 是否传递关键字参数,Paddle 无此参数,暂无转写方式。 |
22-
| always_call | - | 是否强制调用钩子,Paddle 无此参数,暂无转写方式。 |
19+
| hook | hook | 被注册为 forward post-hook 的函数。 |
20+
| always_call | - | 是否强制调用钩子,Paddle 无此参数,一般对训练结果影响不大,可直接删除。 |
21+
22+
### 转写示例
23+
24+
```python
25+
# PyTorch 写法
26+
Linear = torch.nn.Linear(2, 4)
27+
Conv2d = torch.nn.Conv2d(3, 16, 3)
28+
Batch2d = torch.nn.BatchNorm2d(10)
29+
torch.nn.modules.module.register_module_forward_hook(hook)
30+
31+
# Paddle 写法
32+
Linear = paddle.nn.Linear(2, 4)
33+
Conv2d = paddle.nn.Conv2d(3, 16, 3)
34+
Batch2d = paddle.nn.BatchNorm2D(10)
35+
Linear.register_forward_post_hook(hook)
36+
Conv2d.register_forward_post_hook(hook)
37+
Batch2d.register_forward_post_hook(hook)
38+
```

docs/guides/model_convert/convert_from_pytorch/api_difference/nn/torch.nn.modules.module.register_module_forward_pre_hook.md

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
## [ 参数完全一致 ]torch.nn.modules.module.register_module_forward_pre_hook
1+
## [ 组合替代实现 ]torch.nn.modules.module.register_module_forward_pre_hook
22

33
### [torch.nn.modules.module.register_module_forward_pre_hook](https://pytorch.org/docs/stable/generated/torch.nn.modules.module.register_module_forward_pre_hook.html)
44

@@ -12,10 +12,28 @@ torch.nn.modules.module.register_module_forward_pre_hook(hook)
1212
paddle.nn.Layer.register_forward_pre_hook(hook)
1313
```
1414

15-
功能一致,参数完全一致,具体如下:
15+
其中,PyTorch 为给全局所有 module 注册 hook,而 Paddle 为给单个 Layer 注册 hook, 具体如下:
1616

1717
### 参数映射
1818

1919
| PyTorch | PaddlePaddle | 备注 |
2020
|---------|--------------|-----------------------------------------------------------------------------------------------|
2121
| hook | hook | 被注册为 forward pre-hook 的函数。 |
22+
23+
### 转写示例
24+
25+
```python
26+
# PyTorch 写法
27+
Linear = torch.nn.Linear(2, 4)
28+
Conv2d = torch.nn.Conv2d(3, 16, 3)
29+
Batch2d = torch.nn.BatchNorm2d(10)
30+
torch.nn.modules.module.register_module_forward_pre_hook(hook)
31+
32+
# Paddle 写法
33+
Linear = paddle.nn.Linear(2, 4)
34+
Conv2d = paddle.nn.Conv2d(3, 16, 3)
35+
Batch2d = paddle.nn.BatchNorm2D(10)
36+
Linear.register_forward_pre_hook(hook)
37+
Conv2d.register_forward_pre_hook(hook)
38+
Batch2d.register_forward_pre_hook(hook)
39+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_abs
2+
3+
### [torch.\_foreach_abs](https://pytorch.org/docs/stable/generated/torch._foreach_abs.html#torch-foreach-abs)
4+
5+
```python
6+
torch._foreach_abs(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_abs(tensors)
16+
17+
# Paddle 写法
18+
[paddle.abs(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_abs_
2+
3+
### [torch.\_foreach_abs_](https://pytorch.org/docs/stable/generated/torch._foreach_abs_.html#torch-foreach-abs)
4+
5+
```python
6+
torch._foreach_abs_(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_abs_(tensors)
16+
17+
# Paddle 写法
18+
[paddle.abs_(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_acos
2+
3+
### [torch.\_foreach_acos](https://pytorch.org/docs/stable/generated/torch._foreach_acos.html#torch-foreach-acos)
4+
5+
```python
6+
torch._foreach_acos(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_acos(tensors)
16+
17+
# Paddle 写法
18+
[paddle.acos(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_acos_
2+
3+
### [torch.\_foreach_acos_](https://pytorch.org/docs/stable/generated/torch._foreach_acos_.html#torch-foreach-acos)
4+
5+
```python
6+
torch._foreach_acos_(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_acos_(tensors)
16+
17+
# Paddle 写法
18+
[paddle.acos_(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_asin
2+
3+
### [torch.\_foreach_asin](https://pytorch.org/docs/stable/generated/torch._foreach_asin.html#torch-foreach-asin)
4+
5+
```python
6+
torch._foreach_asin(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_asin(tensors)
16+
17+
# Paddle 写法
18+
[paddle.asin(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_asin_
2+
3+
### [torch.\_foreach_asin_](https://pytorch.org/docs/stable/generated/torch._foreach_asin_.html#torch-foreach-asin)
4+
5+
```python
6+
torch._foreach_asin_(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_asin_(tensors)
16+
17+
# Paddle 写法
18+
[paddle.asin_(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_atan
2+
3+
### [torch.\_foreach_atan](https://pytorch.org/docs/stable/generated/torch._foreach_atan.html#torch-foreach-atan)
4+
5+
```python
6+
torch._foreach_atan(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_atan(tensors)
16+
17+
# Paddle 写法
18+
[paddle.atan(x) for x in tensors]
19+
```
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
## [组合替代实现]torch.\_foreach_atan_
2+
3+
### [torch.\_foreach_atan_](https://pytorch.org/docs/stable/generated/torch._foreach_atan_.html#torch-foreach-atan)
4+
5+
```python
6+
torch._foreach_atan_(self)
7+
```
8+
9+
Paddle 无此 API,需要组合实现。
10+
11+
### 转写示例
12+
13+
```python
14+
# PyTorch 写法
15+
torch._foreach_atan_(tensors)
16+
17+
# Paddle 写法
18+
[paddle.atan_(x) for x in tensors]
19+
```

0 commit comments

Comments
 (0)