kutacc_af2_rigid_rot_matmul
Implement matrix-matrix multiplication. The input is two matrices, and the output is a matrix.
Interface Definition
void kutacc_af2_rigid_rot_matmul(kutacc_tensor_h a, kutacc_tensor_h b, kutacc_tensor_h out);
Parameters
Parameter |
Type |
Description |
Input/Output |
tensor shape |
|---|---|---|---|---|
a |
kutacc_tensor_h |
First rotation matrix |
Input |
[..., 3, 3] |
b |
kutacc_tensor_h |
Second rotation matrix |
Input |
[..., 3, 3] |
out |
kutacc_tensor_h |
Output data |
Output |
[..., 3, 3] |
Parameter constraints: dim = a.dim - 2; a.strides[dim] = 3; a.strides[dim +1] = 1; b.strides[dim] = 3; b.strides[dim + 1] = 1
Examples
The rigid operator is used inside the invariant_point_attention operator. For details about the test case, see the example of the invariant_point_attention operator in section "8.1.2.2.2.1.1.9".
Parent topic: AlphaFold2