protovoices-rl-0.1.0.0
Safe HaskellNone
LanguageGHC2021

RL.A2CHelpers

Documentation

type ModelParams (dev :: (DeviceType, Nat)) = Parameters (QModel dev) Source #

type ModelTensors (dev :: (DeviceType, Nat)) = ToModelTensors (ModelParams dev) Source #

newtype UpdateEligCritic Source #

Constructors

UpdateEligCritic QType 

Instances

Instances details
KnownDevice dev => Apply' UpdateEligCritic (QTensor dev shape, QTensor dev shape) (QTensor dev shape) Source # 
Instance details

Defined in RL.A2CHelpers

Methods

apply' :: UpdateEligCritic -> (QTensor dev shape, QTensor dev shape) -> QTensor dev shape #

updateEligCritic :: forall (dev :: (DeviceType, Nat)). KnownDevice dev => QType -> QType -> HList (ModelTensors dev) -> HList (ModelTensors dev) -> HList (ModelTensors dev) Source #

data UpdateEligActor Source #

Constructors

UpdateEligActor QType QType 

Instances

Instances details
KnownDevice dev => Apply' UpdateEligActor (QTensor dev shape, QTensor dev shape) (QTensor dev shape) Source # 
Instance details

Defined in RL.A2CHelpers

Methods

apply' :: UpdateEligActor -> (QTensor dev shape, QTensor dev shape) -> QTensor dev shape #

updateEligActor :: forall (dev :: (DeviceType, Nat)). KnownDevice dev => QType -> QType -> QType -> HList (ModelTensors dev) -> HList (ModelTensors dev) -> HList (ModelTensors dev) Source #

mulModelTensors :: forall (dev :: (DeviceType, Nat)). IsValidDevice dev => QTensor dev ('[] :: [Nat]) -> HList (ModelTensors dev) -> HList (ModelTensors dev) Source #

modelZeros :: forall (dev :: (DeviceType, Nat)). IsValidDevice dev => QModel dev -> HList (ModelTensors dev) Source #

sumTensorList :: forall (dev :: (DeviceType, Nat)). IsValidDevice dev => HList (ModelTensors dev) -> QTensor dev ('[] :: [Nat]) Source #