triton_language.softmax
triton.language.softmax(x, ieee_rounding=False)
计算 x
的逐元素 softmax 函数值。
参数**:**
- x (Block) - 输入值。
这个函数也可作为 tensor
的成员函数调用,使用 x.softmax(...)
而不是 softmax(x, ...)
。
triton.language.softmax(x, ieee_rounding=False)
计算 x
的逐元素 softmax 函数值。
参数**:**
这个函数也可作为 tensor
的成员函数调用,使用 x.softmax(...)
而不是 softmax(x, ...)
。