OpBMMExtensions (dfx)

OpBMMExtensions Module

Type extension	Description
`this.bmm b` Full Usage: `this.bmm b` Parameters: b : `Tensor` - The second tensor. Returns: `Tensor`	Batched matrix product of two tensors. Tensors b must be 3d tensors each containing the same number of matrices. If the tensor is a \(b \times n \times m\) tensor, and b is a \(b \times m \times p\) tensor, the result will be a \(b \times n \times p\) tensor. Extended Type: `Tensor` b : `Tensor` The second tensor. Returns: `Tensor`
`dfx.bmm (a, b)` Full Usage: `dfx.bmm (a, b)` Parameters: a : `Tensor` - The first tensor. b : `Tensor` - The second tensor. Returns: `Tensor`	Batched matrix product of two tensors. Tensors a and b must be 3d tensors each containing the same number of matrices. If a is a \(b \times n \times m\) tensor, b is a \(b \times m \times p\) tensor, the result will be a \(b \times n \times p\) tensor. Extended Type: `dfx` a : `Tensor` The first tensor. b : `Tensor` The second tensor. Returns: `Tensor`