我有一个很大的ndarray X (大概是(1e3,1e3,1e3)),我想做X的操作,包括和不包括第0轴的特定元素(对于第一和第二轴的每个元素)。也就是说,有(1e3,1e3)个元素我想(有时)屏蔽进去或屏蔽出去。# the particular (1e3 x 1e3) elements I'm interested inY = np.ma.masked_array(
DaCe附带了一种语法,允许用户定义一个映射,映射下的是用户为计算而定义的微线程。作为计算,我们可以进行诸如整数加法之类的标量运算。但是,如果我们在微线程中执行张量(矩阵)操作,例如,A@B或A+B,其中A和B是两个矩阵,DaCe将给我们一个错误。 with dace.tasklet:
a << (A+A)[i,