kutacc_af2_rigid_rot_vec_mul

Implement matrix-vector multiplication. The input is a matrix and a vector, and the output is a vector.

void kutacc_af2_rigid_rot_vec_mul(kutacc_tensor_h pts, kutacc_tensor_h rot_mats, kutacc_tensor_h out, kutacc_tensor_h trans);

Parameter	Type	Description	Input/Output	tensor shape
pts	kutacc_tensor_h	Input data	Input	[..., 3]
rot_mats	kutacc_tensor_h	Rotation matrix	Input	[..., 3, 3]
out	kutacc_tensor_h	Output data	Output	[..., 3]
trans	kutacc_tensor_h	Translation vector	Input	[..., 3]

Parameter constraints: dim = pts.dim – 1; pts.strides[dim] = 1; rot_mats.strides[dim] = 3; rot_mats.strides[dim + 1] = 1; trans.strides[dim] = 1

Parent topic: AlphaFold2