如何在gpu上生成python随机数？

Question

我是gpu上python编程的新手。我目前正在尝试使用numbapro.numpy_support.numpy在gpu上生成一组随机数。这是我的代码：

import numpy as np
import numbapro as nbp
from numba.numpy_support import numpy as nbnp

@nbp.vectorize(['float32(int32, int32, float32)'], target='gpu')
def random_syn_weight_mat(n1, n2, prob) :
    A = 1
    A[nbnp.where(A > prob)] = 0
    return A

W = random_syn_weight_mat(1000, 1000, 0.1)

运行此操作会发出错误消息，指出未定义nbnp.where。任何人都可以帮我这个吗？