Mojo function
apply_gumbel_noise_kernel
apply_gumbel_noise_kernel[dtype: DType, input_layout: Layout, num_sms: Int, num_threads: Int](output: LayoutTensor[dtype, input_layout, MutableAnyOrigin], input: LayoutTensor[dtype, input_layout, MutableAnyOrigin], temperature: UnsafePointer[Float32], seed: UnsafePointer[UInt64])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!