Safe Haskell | None |
---|---|
Language | Haskell2010 |
- type family IsValidConcat i (a :: [Nat]) (b :: [Nat]) :: Bool where ...
- type family Squeeze (a :: [Nat]) where ...
- type family Concatenate i (a :: [Nat]) (b :: [Nat]) :: [Nat] where ...
- type family IsValidTranspose (p :: [Nat]) (a :: [Nat]) :: Bool where ...
- type family DimShuffle (a :: [a]) (a :: [Nat]) :: [a] where ...
- dimShuffle :: forall a. Eq a => [a] -> [Nat] -> [a]
- type family Fold i (s :: [Nat]) where ...
- type family FoldAlong i (s :: [Nat]) where ...
- type family TailModule i (s :: [Nat]) where ...
- type family HeadModule i (s :: [Nat]) where ...
- type family Transpose a where ...
Documentation
type family IsValidConcat i (a :: [Nat]) (b :: [Nat]) :: Bool where ... Source #
IsValidConcat _ '[] _ = False | |
IsValidConcat _ _ '[] = False | |
IsValidConcat i a b = And (ZipWith (:==$) (DropDim i a) (DropDim i b)) |
type family DimShuffle (a :: [a]) (a :: [Nat]) :: [a] where ... Source #
DimShuffle _z_6989586621679108036 '[] = '[] | |
DimShuffle '[] _z_6989586621679108039 = '[] | |
DimShuffle ((:) x xs) ((:) b bs) = Case_6989586621679108065 x xs b bs (Let6989586621679108046Scrutinee_6989586621679105664Sym4 x xs b bs) |
dimShuffle :: forall a. Eq a => [a] -> [Nat] -> [a] Source #
type family Fold i (s :: [Nat]) where ... Source #
Reduces axis i in shape s. Does not maintain singlton dimension.
type family FoldAlong i (s :: [Nat]) where ... Source #
Reduces axis i in shape s. Maintains singlton dimension
type family TailModule i (s :: [Nat]) where ... Source #
TailModule _ '[] = '[] | |
TailModule d xs = Snd (SplitAt d xs) |
type family HeadModule i (s :: [Nat]) where ... Source #
HeadModule _ '[] = '[] | |
HeadModule d xs = Fst (SplitAt d xs) |