General dot product of two tensors.
Usage
nvl_dot_general(lhs, rhs, contracting_dims, batching_dims)
Arguments
- lhs, rhs
(tensorish)
Left and right operand.
- contracting_dims
(list())
Dimensions to contract.
- batching_dims
(list())
Batch dimensions.