Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
bb7ca948
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
大约 1 年 前同步成功
通知
328
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
bb7ca948
编写于
11月 01, 2021
作者:
W
WJJ1995
提交者:
GitHub
11月 01, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fixed bug for split (#694)
上级
cfc8c631
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
77 addition
and
1 deletion
+77
-1
docs/pytorch_project_convertor/API_docs/ops/README.md
docs/pytorch_project_convertor/API_docs/ops/README.md
+1
-1
docs/pytorch_project_convertor/API_docs/ops/torch.split.md
docs/pytorch_project_convertor/API_docs/ops/torch.split.md
+76
-0
未找到文件。
docs/pytorch_project_convertor/API_docs/ops/README.md
浏览文件 @
bb7ca948
...
...
@@ -28,7 +28,7 @@
| 23 |
[
torch.narrow
](
https://pytorch.org/docs/stable/generated/torch.narrow.html?highlight=narrow#torch.narrow
)
|
[
paddle.slice
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/slice_cn.html#slice
)
|
[
差异对比
](
https://github.com/PaddlePaddle/X2Paddle/tree/develop/docs/pytorch_project_convertor/API_docs/ops/torch.narrow.md
)
|
| 24 |
[
torch.nonzero
](
https://pytorch.org/docs/stable/generated/torch.nonzero.html?highlight=nonzero#torch.nonzero
)
|
[
paddle.nonzero
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/nonzero_cn.html#nonzero
)
| 功能一致,参数名不一致 |
| 25 |
[
torch.reshape
](
https://pytorch.org/docs/stable/generated/torch.reshape.html?highlight=reshape#torch.reshape
)
|
[
paddle.reshape
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/reshape_cn.html#reshape
)
| 功能一致,参数名不一致 |
| 26 |
[
torch.split
](
https://pytorch.org/docs/stable/generated/torch.split.html?highlight=split#torch.split
)
|
[
paddle.split
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/split_cn.html#split
)
|
功能一致,参数名不一致
|
| 26 |
[
torch.split
](
https://pytorch.org/docs/stable/generated/torch.split.html?highlight=split#torch.split
)
|
[
paddle.split
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/split_cn.html#split
)
|
[
差异对比
](
https://github.com/PaddlePaddle/X2Paddle/tree/develop/docs/pytorch_project_convertor/API_docs/ops/torch.split.md
)
|
| 27 |
[
torch.squeeze
](
https://pytorch.org/docs/stable/generated/torch.squeeze.html?highlight=squeeze#torch.squeeze
)
|
[
paddle.squeeze
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/squeeze_cn.html#squeeze
)
| 功能一致,参数名不一致 |
| 28 |
[
torch.stack
](
https://pytorch.org/docs/stable/generated/torch.stack.html?highlight=stack#torch.stack
)
|
[
paddle.stack
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/stack_cn.html#stack
)
| 功能一致,参数名不一致 |
| 29 |
[
torch.t
](
https://pytorch.org/docs/stable/generated/torch.t.html
)
|
[
paddle.t
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/t_cn.html#t
)
| 功能一致,参数名不一致 |
...
...
docs/pytorch_project_convertor/API_docs/ops/torch.split.md
0 → 100644
浏览文件 @
bb7ca948
## torch.split
### [torch.split](https://pytorch.org/docs/stable/generated/torch.split.html?highlight=torch%20split#torch.split)
```
python
torch
.
split
(
tensor
,
split_size_or_sections
,
dim
=
0
)
```
### [paddle.split](https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/split_cn.html#split)
```
python
paddle
.
split
(
x
,
num_or_sections
,
axis
=
0
,
name
=
None
)
```
### 参数差异
| PyTorch | PaddlePaddle | 备注 |
| ------------- | ------------ | ------------------------------------------------------ |
| tensor | x | 表示输入Tensor。 |
| split_size_or_sections | num_or_sections | 当类型为int时,torch表示单个块大小,paddle表示结果有多少个块 |
| dim | axis | 表示需要分割的维度。 |
### 功能差异
#### 使用方式
***PyTorch**
*
:第二个参数split_size_or_sections类型为int或者list(int)。
***PaddlePaddle**
*
:第二个参数num_or_sections类型为int、list(int)或者tuple(int)。
### 代码示例
```
python
# PyTorch示例:
a
=
torch
.
arange
(
10
).
reshape
(
5
,
2
)
# 输出
# tensor([[0, 1],
# [2, 3],
# [4, 5],
# [6, 7],
# [8, 9]])
torch
.
split
(
a
,
2
,
1
)
# 输出
# (tensor([[0, 1],
# [2, 3],
# [4, 5],
# [6, 7],
# [8, 9]]),)
```
```
python
# PaddlePaddle示例:
b
=
paddle
.
arange
(
10
).
reshape
([
5
,
2
])
# 输出
# Tensor(shape=[5, 2], dtype=int64, place=CPUPlace, stop_gradient=True,
# [[0, 1],
# [2, 3],
# [4, 5],
# [6, 7],
# [8, 9]])
paddle
.
split
(
b
,
2
,
1
)
# 输出
# [Tensor(shape=[5, 1], dtype=int64, place=CPUPlace, stop_gradient=True,
# [[0],
# [2],
# [4],
# [6],
# [8]]), Tensor(shape=[5, 1], dtype=int64, place=CPUPlace, stop_gradient=True,
# [[1],
# [3],
# [5],
# [7],
# [9]])]
```
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录