pythonの関数の書き方についてご教示いただけませんでしょうか?
from numba import cuda import numpy as np @cuda.jit def increment_by_one(arr): tx = cuda.threadIdx.x ty = cuda.blockIdx.x bw = cuda.blockDim.x pos = tx + ty * bw arr[pos] += 1 arr = np.arange(5) print('Input: {}'.format(arr)) threadsperblock = 32 blockspergrid = 1 increment_by_one[blockspergrid, threadsperblock](arr) print('Output: {}'.format(arr))
このコードの
increment_by_one[blockspergrid, threadsperblock](arr)
この部分が理解できません。
increment_by_one(arr) これなら理解できるのですが・・・
[blockspergrid, threadsperblock] これが引数の前に入ることでどのような動きになるのでしょうか?そもそも構文として許されるのでしょうか?
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2019/01/16 17:20 編集
2019/01/16 17:23
2019/01/16 22:50