get_direction_vectors32#

nvmath.device.random_helpers.get_direction_vectors32(int set_, size_t size)[source]#

获取用于 32 位准随机数生成的方向向量。

参数:
  • set (DirectionVectorSet) – 要使用的方向向量集。

  • size (size_t) – 维度数量。

返回:

形状为 (size, 32) 的数组。

返回类型:

numpy.ndarray

另请参阅

curandGetDirectionVectors32