28 lines
604 B
Rust
28 lines
604 B
Rust
//@ known-bug: #115808
|
|
#![feature(generic_const_exprs)]
|
|
|
|
use std::ops::Mul;
|
|
|
|
pub trait Indices<const N: usize> {
|
|
const NUM_ELEMS: usize;
|
|
}
|
|
|
|
pub trait Concat<J> {
|
|
type Output;
|
|
}
|
|
|
|
pub struct Tensor<I: Indices<N>, const N: usize>
|
|
where
|
|
[u8; I::NUM_ELEMS]: Sized, {}
|
|
|
|
impl<I: Indices<N>, J: Indices<N>, const N: usize> Mul<Tensor<J, N>> for Tensor<I, N>
|
|
where
|
|
I: Concat<FN>,
|
|
<I as Concat<J>>::Output: Indices<N>,
|
|
[u8; I::NUM_ELEMS]: Sized,
|
|
[u8; J::NUM_ELEMS]: Sized,
|
|
[u8; <I as Concat<J>>::Output::NUM_ELEMS]: Sized,
|
|
{
|
|
type Output = Tensor<<I as Concat<J>>::Output, N>;
|
|
}
|