Cupy中的就地减法与广播

Posted 2023-02-16

【中文标题】Cupy中的就地减法与广播

【英文标题】：in-place subtraction in Cupy with broadcasting

【发布时间】：2021-05-30 10:03:28

【问题描述】：

假设我想这样做：

a=cp.zeros((10,10,50))
b=cp.ones((10,10,1))
a[:]=a-b

是否可以在不使用额外内存的情况下进行减法运算？

【参考方案1】：

这是所有 Python 库（如果支持）中的有效就地操作，包括 NumPy 和 CuPy：

a -= b

【讨论】：

感谢您的回答，但我相信所涉及的广播会导致cupy中额外的内存使用，至少看起来是因为我在执行时遇到内存错误。应该可以在不使用额外内存的情况下执行此操作