triton.language.dot¶
- triton.language.dot(input, other, trans_a=False, trans_b=False, allow_tf32=True)¶
Returns the matrix product of two blocks.
The two blocks must be two dimensionals and have compatible inner dimensions.
- Parameters
input (2D tensor of scalar-type in {
float16,bfloat16,float32}) – The first tensor to be multiplied.other (2D tensor of scalar-type in {
float16,bfloat16,float32}) – The second tensor to be multiplied.