【问题标题】:in-place subtraction in Cupy with broadcastingCupy中的就地减法与广播
【发布时间】:2021-05-30 10:03:28
【问题描述】:

假设我想这样做:

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

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

【问题讨论】:

    标签: cupy


    【解决方案1】:

    这是所有 Python 库(如果支持)中的有效就地操作,包括 NumPy 和 CuPy:

    a -= b
    

    【讨论】:

    • 感谢您的回答,但我相信所涉及的广播会导致cupy中额外的内存使用,至少看起来是因为我在执行时遇到内存错误。应该可以在不使用额外内存的情况下执行此操作
    猜你喜欢
    • 2014-05-17
    • 1970-01-01
    • 1970-01-01
    • 2021-05-22
    • 2013-05-15
    • 2011-02-28
    • 2019-03-08
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多