Skip to content

【PaddlePaddle Hackathon 2】11、在 Paddle 中新增 paddle.index_fill 功能 #40324

Open
@TCChenlong

Description

@TCChenlong

(此 ISSUE 为 PaddlePaddle Hackathon 第二期活动的任务 ISSUE,更多详见 【PaddlePaddle Hackathon 第二期】任务总览

【任务说明】

任务标题:为 Paddle 新增 index_fill API

技术标签:深度学习框架,Python

任务难度:困难

详细描述:对于 nd tensor, 沿着某个轴 axis 取 (n-1)d 的切片,索引位置是 index, 并且将 value 中值填充到这些切片上。其中 value 是一个 scalar 或者 0d tensor, 该运算需要支持微分。调用路径:

  1. paddle.index_fill 作为独立的函数调用,非 inplace
  2. paddle.index_fill_,作为独立的函数,inplace 地修改输入;
  3. Tensor.index_fill(axis, index, value) 做为 Tensor 的方法使用,非 inplace;
  4. Tensor.index_fill_(axis, index, value) 做为 Tensor 的方法使用, inplace 修改输入;

【提交流程】

请按 新增API 开发&提交流程中的描述,完成任务提交过程中的相关步骤。

【提交内容】

【合入标准】

【技术要求】

  • 了解 Paddle 算子开发流程
  • 熟练掌握 C++,Python
  • 有一定的 CUDA 开发经验

【参考内容】

【答疑交流】

  • 如果在开发中对于上述任务有任何问题,欢迎在本 ISSUE 下留言交流。
  • 对于开发中的共性问题,在活动过程中,会定期组织答疑,请大家关注官网&QQ群的通知,及时参与。

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions