| Safe Haskell | None |
|---|---|
| Language | GHC2021 |
RL.ModelTypes
Contents
Documentation
type IsValidDevice (dev :: (DeviceType, Nat)) = (GeluDTypeIsValid dev QDType, RandDTypeIsValid dev QDType, BasicArithmeticDTypeIsValid dev QDType, SumDTypeIsValid dev QDType, MeanDTypeValidation dev QDType, StandardFloatingPointDTypeValidation dev QDType, KnownDevice dev) Source #
opts :: forall (dev :: (DeviceType, Nat)). KnownDevice dev => TensorOptions Source #
toOpts :: forall (dev :: (DeviceType, Nat)) a. (KnownDevice dev, HasTypes a Tensor) => a -> a Source #
toQTensor' :: forall (dev :: (DeviceType, Nat)). KnownDevice dev => QType -> Tensor Source #
toQTensor :: forall (dev :: (DeviceType, Nat)). KnownDevice dev => QType -> QTensor dev ('[] :: [Nat]) Source #
type MaxPitches = 8 Source #
type PVAction = Action (Notes SPitch) (Edges SPitch) (Split SPitch) (Freeze SPitch) (Spread SPitch) Source #
type PVActionResult = Either PVState (Edges SPitch, [PVLeftmost SPitch]) Source #
type PVRewardFn label = PVActionResult -> Maybe (NonEmpty PVAction) -> PVAction -> label -> IO QType Source #
type CommonHiddenSize = 8 Source #
type FifthPadding = 6 Source #
type OctavePadding = 2 Source #
type EmbSize = CommonHiddenSize Source #
type OctaveSize = (2 * OctavePadding) + 1 Source #
type PShape = '[FifthSize, OctaveSize] Source #
type QOutHidden = CommonHiddenSize Source #
type QSliceHidden = CommonHiddenSize Source #
type QTransHidden = CommonHiddenSize Source #
type QActionHidden = CommonHiddenSize Source #
type QStateHidden = CommonHiddenSize Source #