get_direction_vectors32#
-
nvmath.
device. get_direction_vectors32(int set_, size_t size)[source]#random_helpers. 获取用于 32 位准随机数生成的方向向量。
- 参数:
set (DirectionVectorSet) – 要使用的方向向量集。
size (size_t) – 维度数量。
- 返回:
形状为
(size, 32)
的数组。- 返回类型:
另请参阅
curandGetDirectionVectors32