【发布时间】:2021-05-30 10:03:28
【问题描述】:
假设我想这样做:
a=cp.zeros((10,10,50))
b=cp.ones((10,10,1))
a[:]=a-b
是否可以在不使用额外内存的情况下进行减法运算?
【问题讨论】:
标签: cupy
假设我想这样做:
a=cp.zeros((10,10,50))
b=cp.ones((10,10,1))
a[:]=a-b
是否可以在不使用额外内存的情况下进行减法运算?
【问题讨论】:
标签: cupy
这是所有 Python 库(如果支持)中的有效就地操作,包括 NumPy 和 CuPy:
a -= b
【讨论】: