Trait dfdx::tensor_ops::MatMulTrTyping

source · [−]

pub trait MatMulTrTyping<B> {
    type C;
}

Expand description

Enables concrete output types for generic matmul functions. Without this you’d have to specify type of output.

Required Associated Types

type C

Implementors

impl<const B1: usize, const B2: usize, const M: usize, const N: usize, const K: usize, H> MatMulTrTyping<Tensor4D<B1, B2, N, K, NoneTape>> for Tensor4D<B1, B2, M, K, H>

type C = Tensor4D<B1, B2, M, N, H>

impl<const B: usize, const M: usize, const N: usize, const K: usize, H> MatMulTrTyping<Tensor2D<N, K, NoneTape>> for Tensor3D<B, M, K, H>

type C = Tensor3D<B, M, N, H>

impl<const B: usize, const M: usize, const N: usize, const K: usize, H> MatMulTrTyping<Tensor3D<B, N, K, NoneTape>> for Tensor3D<B, M, K, H>

type C = Tensor3D<B, M, N, H>

impl<const M: usize, const N: usize, const K: usize, H> MatMulTrTyping<Tensor2D<N, K, NoneTape>> for Tensor2D<M, K, H>

type C = Tensor2D<M, N, H>