hasktorch-0.2.1.3: Haskell bindings to libtorch, supporting both typed and untyped tensors.
Safe HaskellNone
LanguageHaskell2010

Torch.Typed.Parameter

Documentation

newtype Parameter (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) Source #

Constructors

UnsafeMkParameter IndependentTensor 

Instances

Instances details
KnownDType dtype' => HasToDType (dtype' :: DType) (dtype :: DType) (Parameter device dtype shape) (Parameter device dtype' shape) Source # 
Instance details

Defined in Torch.Typed.DType

Methods

toDType :: Parameter device dtype shape -> Parameter device dtype' shape Source #

KnownDevice device' => HasToDevice device' device (Parameter device dtype shape) (Parameter device' dtype shape) Source # 
Instance details

Defined in Torch.Typed.Device

Methods

toDevice :: Parameter device dtype shape -> Parameter device' dtype shape Source #

Apply' MakeIndependent (Tensor device dtype shape) (IO (Parameter device dtype shape)) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

apply' :: MakeIndependent -> Tensor device dtype shape -> IO (Parameter device dtype shape) Source #

Apply' ToParameter (Tensor dev dtype shape) (Parameter dev dtype shape) Source # 
Instance details

Defined in Torch.Typed.Optim.CppOptim

Methods

apply' :: ToParameter -> Tensor dev dtype shape -> Parameter dev dtype shape Source #

Apply' ToDependent (Parameter device dtype shape) (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

apply' :: ToDependent -> Parameter device dtype shape -> Tensor device dtype shape Source #

Show (Parameter device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

showsPrec :: Int -> Parameter device dtype shape -> ShowS #

show :: Parameter device dtype shape -> String #

showList :: [Parameter device dtype shape] -> ShowS #

Parameterized (Parameter device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters (Parameter device dtype shape) 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (Parameter device dtype shape) = '[Parameter device dtype shape]

Methods

flattenParameters :: Parameter device dtype shape -> HList (Parameters (Parameter device dtype shape)) Source #

replaceParameters :: Parameter device dtype shape -> HList (Parameters (Parameter device dtype shape)) -> Parameter device dtype shape Source #

HasGrad (Parameter device dtype shape) (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Autograd

Methods

grad :: forall (dtype0 :: DType) (device0 :: (DeviceType, Nat)). Tensor device0 dtype0 ('[] :: [Nat]) -> Parameter device dtype shape -> Tensor device dtype shape Source #

toDependent :: Parameter device dtype shape -> Tensor device dtype shape

type Parameters (Parameter device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (Parameter device dtype shape) = '[Parameter device dtype shape]

parameterToDType :: forall (dtype' :: DType) (dtype :: DType) (device :: (DeviceType, Nat)) (shape :: [Nat]). KnownDType dtype' => Parameter device dtype shape -> Parameter device dtype' shape Source #

parameterToDevice :: forall (device' :: (DeviceType, Nat)) (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). KnownDevice device' => Parameter device dtype shape -> Parameter device' dtype shape Source #

makeIndependent :: forall (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). Tensor device dtype shape -> IO (Parameter device dtype shape) Source #

toDependent :: forall (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). Parameter device dtype shape -> Tensor device dtype shape Source #

class Parameterized f where Source #

Minimal complete definition

Nothing

Associated Types

type Parameters f :: [Type] Source #

type Parameters f = GParameters (Rep f)

Methods

flattenParameters :: f -> HList (Parameters f) Source #

default flattenParameters :: (Generic f, GParameterized (Rep f), Parameters f ~ GParameters (Rep f)) => f -> HList (Parameters f) Source #

replaceParameters :: f -> HList (Parameters f) -> f Source #

default replaceParameters :: (Generic f, GParameterized (Rep f), Parameters f ~ GParameters (Rep f)) => f -> HList (Parameters f) -> f Source #

Instances

Instances details
Parameterized Dropout Source # 
Instance details

Defined in Torch.Typed.NN.Dropout

Associated Types

type Parameters Dropout 
Instance details

Defined in Torch.Typed.NN.Dropout

type Parameters Dropout = GParameters (Rep Dropout)

Methods

flattenParameters :: Dropout -> HList (Parameters Dropout) Source #

replaceParameters :: Dropout -> HList (Parameters Dropout) -> Dropout Source #

Parameterized GD Source # 
Instance details

Defined in Torch.Typed.Optim

Associated Types

type Parameters GD 
Instance details

Defined in Torch.Typed.Optim

type Parameters GD = '[] :: [Type]

Methods

flattenParameters :: GD -> HList (Parameters GD) Source #

replaceParameters :: GD -> HList (Parameters GD) -> GD Source #

Parameterized Double Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters Double 
Instance details

Defined in Torch.Typed.Parameter

type Parameters Double = '[] :: [Type]

Methods

flattenParameters :: Double -> HList (Parameters Double) Source #

replaceParameters :: Double -> HList (Parameters Double) -> Double Source #

Parameterized Float Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters Float 
Instance details

Defined in Torch.Typed.Parameter

type Parameters Float = '[] :: [Type]

Methods

flattenParameters :: Float -> HList (Parameters Float) Source #

replaceParameters :: Float -> HList (Parameters Float) -> Float Source #

Parameterized Int Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters Int 
Instance details

Defined in Torch.Typed.Parameter

type Parameters Int = '[] :: [Type]

Methods

flattenParameters :: Int -> HList (Parameters Int) Source #

replaceParameters :: Int -> HList (Parameters Int) -> Int Source #

HAppendFD momenta momenta (momenta ++ momenta) => Parameterized (Adam momenta) Source # 
Instance details

Defined in Torch.Typed.Optim

Associated Types

type Parameters (Adam momenta) 
Instance details

Defined in Torch.Typed.Optim

type Parameters (Adam momenta) = AdamIter ': (momenta ++ momenta)

Methods

flattenParameters :: Adam momenta -> HList (Parameters (Adam momenta)) Source #

replaceParameters :: Adam momenta -> HList (Parameters (Adam momenta)) -> Adam momenta Source #

Parameterized (GDM momenta) Source # 
Instance details

Defined in Torch.Typed.Optim

Associated Types

type Parameters (GDM momenta) 
Instance details

Defined in Torch.Typed.Optim

type Parameters (GDM momenta) = momenta

Methods

flattenParameters :: GDM momenta -> HList (Parameters (GDM momenta)) Source #

replaceParameters :: GDM momenta -> HList (Parameters (GDM momenta)) -> GDM momenta Source #

(Parameterized f, Parameterized (HList fs), HAppendFD (Parameters f) (Parameters (HList fs)) (Parameters f ++ Parameters (HList fs))) => Parameterized (HList (f ': fs)) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters (HList (f ': fs)) 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (HList (f ': fs)) = Parameters f ++ Parameters (HList fs)

Methods

flattenParameters :: HList (f ': fs) -> HList (Parameters (HList (f ': fs))) Source #

replaceParameters :: HList (f ': fs) -> HList (Parameters (HList (f ': fs))) -> HList (f ': fs) Source #

Parameterized (HList ('[] :: [k])) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters (HList ('[] :: [k])) 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (HList ('[] :: [k])) = '[] :: [Type]

Methods

flattenParameters :: HList ('[] :: [k]) -> HList (Parameters (HList ('[] :: [k]))) Source #

replaceParameters :: HList ('[] :: [k]) -> HList (Parameters (HList ('[] :: [k]))) -> HList ('[] :: [k]) Source #

Parameterized (LayerNorm normalizedShape dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Normalization

Associated Types

type Parameters (LayerNorm normalizedShape dtype device) 
Instance details

Defined in Torch.Typed.NN.Normalization

type Parameters (LayerNorm normalizedShape dtype device) = GParameters (Rep (LayerNorm normalizedShape dtype device))

Methods

flattenParameters :: LayerNorm normalizedShape dtype device -> HList (Parameters (LayerNorm normalizedShape dtype device)) Source #

replaceParameters :: LayerNorm normalizedShape dtype device -> HList (Parameters (LayerNorm normalizedShape dtype device)) -> LayerNorm normalizedShape dtype device Source #

Parameterized (Parameter device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters (Parameter device dtype shape) 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (Parameter device dtype shape) = '[Parameter device dtype shape]

Methods

flattenParameters :: Parameter device dtype shape -> HList (Parameters (Parameter device dtype shape)) Source #

replaceParameters :: Parameter device dtype shape -> HList (Parameters (Parameter device dtype shape)) -> Parameter device dtype shape Source #

Parameterized (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type Parameters (Tensor device dtype shape) 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (Tensor device dtype shape) = '[] :: [Type]

Methods

flattenParameters :: Tensor device dtype shape -> HList (Parameters (Tensor device dtype shape)) Source #

replaceParameters :: Tensor device dtype shape -> HList (Parameters (Tensor device dtype shape)) -> Tensor device dtype shape Source #

Parameterized (Linear inputFeatures outputFeatures dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Linear

Associated Types

type Parameters (Linear inputFeatures outputFeatures dtype device) 
Instance details

Defined in Torch.Typed.NN.Linear

type Parameters (Linear inputFeatures outputFeatures dtype device) = GParameters (Rep (Linear inputFeatures outputFeatures dtype device))

Methods

flattenParameters :: Linear inputFeatures outputFeatures dtype device -> HList (Parameters (Linear inputFeatures outputFeatures dtype device)) Source #

replaceParameters :: Linear inputFeatures outputFeatures dtype device -> HList (Parameters (Linear inputFeatures outputFeatures dtype device)) -> Linear inputFeatures outputFeatures dtype device Source #

Parameterized (GRUCell inputDim hiddenDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Associated Types

type Parameters (GRUCell inputDim hiddenDim dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

type Parameters (GRUCell inputDim hiddenDim dtype device) = GParameters (Rep (GRUCell inputDim hiddenDim dtype device))

Methods

flattenParameters :: GRUCell inputDim hiddenDim dtype device -> HList (Parameters (GRUCell inputDim hiddenDim dtype device)) Source #

replaceParameters :: GRUCell inputDim hiddenDim dtype device -> HList (Parameters (GRUCell inputDim hiddenDim dtype device)) -> GRUCell inputDim hiddenDim dtype device Source #

Parameterized (LSTMCell inputDim hiddenDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.LSTM

Associated Types

type Parameters (LSTMCell inputDim hiddenDim dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.LSTM

type Parameters (LSTMCell inputDim hiddenDim dtype device) = GParameters (Rep (LSTMCell inputDim hiddenDim dtype device))

Methods

flattenParameters :: LSTMCell inputDim hiddenDim dtype device -> HList (Parameters (LSTMCell inputDim hiddenDim dtype device)) Source #

replaceParameters :: LSTMCell inputDim hiddenDim dtype device -> HList (Parameters (LSTMCell inputDim hiddenDim dtype device)) -> LSTMCell inputDim hiddenDim dtype device Source #

Parameterized (TransformerMLP embedDim ffnDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Associated Types

type Parameters (TransformerMLP embedDim ffnDim dtype device) 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (TransformerMLP embedDim ffnDim dtype device) = GParameters (Rep (TransformerMLP embedDim ffnDim dtype device))

Methods

flattenParameters :: TransformerMLP embedDim ffnDim dtype device -> HList (Parameters (TransformerMLP embedDim ffnDim dtype device)) Source #

replaceParameters :: TransformerMLP embedDim ffnDim dtype device -> HList (Parameters (TransformerMLP embedDim ffnDim dtype device)) -> TransformerMLP embedDim ffnDim dtype device Source #

Parameterized (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) = GParameters (Rep (Conv1d inputChannelSize outputChannelSize kernelSize dtype device))

Methods

flattenParameters :: Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> HList (Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device)) Source #

replaceParameters :: Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> HList (Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device)) -> Conv1d inputChannelSize outputChannelSize kernelSize dtype device Source #

Parameterized (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) = GParameters (Rep (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device))

Methods

flattenParameters :: ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> HList (Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device)) Source #

replaceParameters :: ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> HList (Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device)) -> ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device Source #

Parameterized (GRULayer inputSize hiddenSize 'Bidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Associated Types

type Parameters (GRULayer inputSize hiddenSize 'Bidirectional dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRULayer inputSize hiddenSize 'Bidirectional dtype device) = '[Parameter device dtype (GRUWIShape hiddenSize inputSize), Parameter device dtype (GRUWHShape hiddenSize inputSize), Parameter device dtype (GRUBIShape hiddenSize inputSize), Parameter device dtype (GRUBHShape hiddenSize inputSize), Parameter device dtype (GRUWIShape hiddenSize inputSize), Parameter device dtype (GRUWHShape hiddenSize inputSize), Parameter device dtype (GRUBIShape hiddenSize inputSize), Parameter device dtype (GRUBHShape hiddenSize inputSize)]

Methods

flattenParameters :: GRULayer inputSize hiddenSize 'Bidirectional dtype device -> HList (Parameters (GRULayer inputSize hiddenSize 'Bidirectional dtype device)) Source #

replaceParameters :: GRULayer inputSize hiddenSize 'Bidirectional dtype device -> HList (Parameters (GRULayer inputSize hiddenSize 'Bidirectional dtype device)) -> GRULayer inputSize hiddenSize 'Bidirectional dtype device Source #

Parameterized (GRULayer inputSize hiddenSize 'Unidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Associated Types

type Parameters (GRULayer inputSize hiddenSize 'Unidirectional dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRULayer inputSize hiddenSize 'Unidirectional dtype device) = '[Parameter device dtype (GRUWIShape hiddenSize inputSize), Parameter device dtype (GRUWHShape hiddenSize inputSize), Parameter device dtype (GRUBIShape hiddenSize inputSize), Parameter device dtype (GRUBHShape hiddenSize inputSize)]

Methods

flattenParameters :: GRULayer inputSize hiddenSize 'Unidirectional dtype device -> HList (Parameters (GRULayer inputSize hiddenSize 'Unidirectional dtype device)) Source #

replaceParameters :: GRULayer inputSize hiddenSize 'Unidirectional dtype device -> HList (Parameters (GRULayer inputSize hiddenSize 'Unidirectional dtype device)) -> GRULayer inputSize hiddenSize 'Unidirectional dtype device Source #

Parameterized (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Associated Types

type Parameters (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) = '[Parameter device dtype (LSTMWIShape hiddenSize inputSize), Parameter device dtype (LSTMWHShape hiddenSize inputSize), Parameter device dtype (LSTMBIShape hiddenSize inputSize), Parameter device dtype (LSTMBHShape hiddenSize inputSize), Parameter device dtype (LSTMWIShape hiddenSize inputSize), Parameter device dtype (LSTMWHShape hiddenSize inputSize), Parameter device dtype (LSTMBIShape hiddenSize inputSize), Parameter device dtype (LSTMBHShape hiddenSize inputSize)]

Methods

flattenParameters :: LSTMLayer inputSize hiddenSize 'Bidirectional dtype device -> HList (Parameters (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device)) Source #

replaceParameters :: LSTMLayer inputSize hiddenSize 'Bidirectional dtype device -> HList (Parameters (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device)) -> LSTMLayer inputSize hiddenSize 'Bidirectional dtype device Source #

Parameterized (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Associated Types

type Parameters (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) = '[Parameter device dtype (LSTMWIShape hiddenSize inputSize), Parameter device dtype (LSTMWHShape hiddenSize inputSize), Parameter device dtype (LSTMBIShape hiddenSize inputSize), Parameter device dtype (LSTMBHShape hiddenSize inputSize)]

Methods

flattenParameters :: LSTMLayer inputSize hiddenSize 'Unidirectional dtype device -> HList (Parameters (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device)) Source #

replaceParameters :: LSTMLayer inputSize hiddenSize 'Unidirectional dtype device -> HList (Parameters (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device)) -> LSTMLayer inputSize hiddenSize 'Unidirectional dtype device Source #

Parameterized (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) = GParameters (Rep (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device))

Methods

flattenParameters :: Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> HList (Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device)) Source #

replaceParameters :: Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> HList (Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device)) -> Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device Source #

Parameterized (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) = GParameters (Rep (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device))

Methods

flattenParameters :: ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> HList (Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device)) Source #

replaceParameters :: ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> HList (Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device)) -> ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device Source #

(1 <= numLayers, Parameterized (GRULayerStack inputSize hiddenSize numLayers directionality dtype device), HAppendFD (Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device)) (Parameters Dropout) (Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) ++ Parameters Dropout)) => Parameterized (GRU inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Associated Types

type Parameters (GRU inputSize hiddenSize numLayers directionality dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRU inputSize hiddenSize numLayers directionality dtype device) = GParameters (Rep (GRU inputSize hiddenSize numLayers directionality dtype device))

Methods

flattenParameters :: GRU inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (GRU inputSize hiddenSize numLayers directionality dtype device)) Source #

replaceParameters :: GRU inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (GRU inputSize hiddenSize numLayers directionality dtype device)) -> GRU inputSize hiddenSize numLayers directionality dtype device Source #

(1 <= numLayers, (2 <=? numLayers) ~ flag, GRULayerStackParameterized flag inputSize hiddenSize numLayers directionality dtype device) => Parameterized (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Associated Types

type Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) = GRULayerStackParameters (2 <=? numLayers) inputSize hiddenSize numLayers directionality dtype device

Methods

flattenParameters :: GRULayerStack inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device)) Source #

replaceParameters :: GRULayerStack inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device)) -> GRULayerStack inputSize hiddenSize numLayers directionality dtype device Source #

(1 <= numLayers, Parameterized (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device), HAppendFD (Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device)) (Parameters Dropout) (Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) ++ Parameters Dropout)) => Parameterized (LSTM inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Associated Types

type Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device) = GParameters (Rep (LSTM inputSize hiddenSize numLayers directionality dtype device))

Methods

flattenParameters :: LSTM inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device)) Source #

replaceParameters :: LSTM inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device)) -> LSTM inputSize hiddenSize numLayers directionality dtype device Source #

(1 <= numLayers, (2 <=? numLayers) ~ flag, LSTMLayerStackParameterized flag inputSize hiddenSize numLayers directionality dtype device) => Parameterized (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Associated Types

type Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) = LSTMLayerStackParameters (2 <=? numLayers) inputSize hiddenSize numLayers directionality dtype device

Methods

flattenParameters :: LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device)) Source #

replaceParameters :: LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device -> HList (Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device)) -> LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device Source #

Parameterized (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

Associated Types

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) 
Instance details

Defined in Torch.Typed.NN.Sparse

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) = GParameters (Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device))

Methods

flattenParameters :: Embedding paddingIdx numEmbeds embedSize 'Constant dtype device -> HList (Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device)) Source #

replaceParameters :: Embedding paddingIdx numEmbeds embedSize 'Constant dtype device -> HList (Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device)) -> Embedding paddingIdx numEmbeds embedSize 'Constant dtype device Source #

Parameterized (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

Associated Types

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) 
Instance details

Defined in Torch.Typed.NN.Sparse

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) = GParameters (Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device))

Methods

flattenParameters :: Embedding paddingIdx numEmbeds embedSize 'Learned dtype device -> HList (Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device)) Source #

replaceParameters :: Embedding paddingIdx numEmbeds embedSize 'Learned dtype device -> HList (Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device)) -> Embedding paddingIdx numEmbeds embedSize 'Learned dtype device Source #

Parameterized (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Associated Types

type Parameters (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) = GParameters (Rep (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device))

Methods

flattenParameters :: MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device -> HList (Parameters (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device)) Source #

replaceParameters :: MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device -> HList (Parameters (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device)) -> MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device Source #

Parameterized (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) = GParameters (Rep (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device))

Methods

flattenParameters :: Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> HList (Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device)) Source #

replaceParameters :: Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> HList (Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device)) -> Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device Source #

Parameterized (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) = GParameters (Rep (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device))

Methods

flattenParameters :: ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> HList (Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device)) Source #

replaceParameters :: ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> HList (Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device)) -> ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device Source #

(Parameterized (GRU inputSize hiddenSize numLayers directionality dtype device), HAppendFD (Parameters (GRU inputSize hiddenSize numLayers directionality dtype device)) ('[] :: [Type]) (Parameters (GRU inputSize hiddenSize numLayers directionality dtype device) ++ ('[] :: [Type]))) => Parameterized (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Associated Types

type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) = GParameters (Rep (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device))

Methods

flattenParameters :: GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device -> HList (Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device)) Source #

replaceParameters :: GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device -> HList (Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device)) -> GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device Source #

(Parameterized (GRU inputSize hiddenSize numLayers directionality dtype device), HAppendFD (Parameters (GRU inputSize hiddenSize numLayers directionality dtype device)) '[Parameter device dtype '[numLayers * NumberOfDirections directionality, hiddenSize]] (Parameters (GRU inputSize hiddenSize numLayers directionality dtype device) ++ '[Parameter device dtype '[numLayers * NumberOfDirections directionality, hiddenSize]])) => Parameterized (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Associated Types

type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) = GParameters (Rep (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device))

Methods

flattenParameters :: GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device -> HList (Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device)) Source #

replaceParameters :: GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device -> HList (Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device)) -> GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device Source #

(Parameterized (LSTM inputSize hiddenSize numLayers directionality dtype device), HAppendFD (Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device)) ('[] :: [Type]) (Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device) ++ ('[] :: [Type]))) => Parameterized (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Associated Types

type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) = GParameters (Rep (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device))

Methods

flattenParameters :: LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device -> HList (Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device)) Source #

replaceParameters :: LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device -> HList (Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device)) -> LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device Source #

(Parameterized (LSTM inputSize hiddenSize numLayers directionality dtype device), HAppendFD (Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device)) '[Parameter device dtype '[numLayers * NumberOfDirections directionality, hiddenSize], Parameter device dtype '[numLayers * NumberOfDirections directionality, hiddenSize]] (Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device) ++ '[Parameter device dtype '[numLayers * NumberOfDirections directionality, hiddenSize], Parameter device dtype '[numLayers * NumberOfDirections directionality, hiddenSize]])) => Parameterized (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Associated Types

type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) = GParameters (Rep (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device))

Methods

flattenParameters :: LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device -> HList (Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device)) Source #

replaceParameters :: LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device -> HList (Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device)) -> LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device Source #

Parameterized (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Associated Types

type Parameters (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) = GParameters (Rep (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device))

Methods

flattenParameters :: TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device -> HList (Parameters (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device)) Source #

replaceParameters :: TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device -> HList (Parameters (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device)) -> TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device Source #

(layers ~ HReplicateR numAttnLayers (TransformerLayer embedDim embedDim embedDim numHeads ffnDim dtype device), Parameterized (HList layers), HAppendFD (Parameters (HList layers)) '[Parameter device dtype '[numEmbeds, embedDim], Parameter device dtype '[numEmbeds]] (Parameters (HList layers) ++ '[Parameter device dtype '[numEmbeds, embedDim], Parameter device dtype '[numEmbeds]])) => Parameterized (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Associated Types

type Parameters (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) = GParameters (Rep (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device))

Methods

flattenParameters :: TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device -> HList (Parameters (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device)) Source #

replaceParameters :: TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device -> HList (Parameters (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device)) -> TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device Source #

class GParameterized (f :: Type -> Type) where Source #

Associated Types

type GParameters (f :: Type -> Type) :: [Type] Source #

Methods

gFlattenParameters :: f a -> HList (GParameters f) Source #

gReplaceParameters :: f a -> HList (GParameters f) -> f a Source #

Instances

Instances details
GParameterized (U1 :: Type -> Type) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type GParameters (U1 :: Type -> Type) 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (U1 :: Type -> Type) = '[] :: [Type]

Methods

gFlattenParameters :: U1 a -> HList (GParameters (U1 :: Type -> Type)) Source #

gReplaceParameters :: U1 a -> HList (GParameters (U1 :: Type -> Type)) -> U1 a Source #

(GParameterized l, GParameterized r, HAppendFD (GParameters l) (GParameters r) (GParameters l ++ GParameters r)) => GParameterized (l :*: r) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type GParameters (l :*: r) 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (l :*: r) = GParameters l ++ GParameters r

Methods

gFlattenParameters :: (l :*: r) a -> HList (GParameters (l :*: r)) Source #

gReplaceParameters :: (l :*: r) a -> HList (GParameters (l :*: r)) -> (l :*: r) a Source #

Parameterized f => GParameterized (K1 i f :: Type -> Type) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type GParameters (K1 i f :: Type -> Type) 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (K1 i f :: Type -> Type) = Parameters f

Methods

gFlattenParameters :: K1 i f a -> HList (GParameters (K1 i f :: Type -> Type)) Source #

gReplaceParameters :: K1 i f a -> HList (GParameters (K1 i f :: Type -> Type)) -> K1 i f a Source #

GParameterized f => GParameterized (M1 i t f) Source # 
Instance details

Defined in Torch.Typed.Parameter

Associated Types

type GParameters (M1 i t f) 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (M1 i t f) = GParameters f

Methods

gFlattenParameters :: M1 i t f a -> HList (GParameters (M1 i t f)) Source #

gReplaceParameters :: M1 i t f a -> HList (GParameters (M1 i t f)) -> M1 i t f a Source #

untypeParam :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). Parameter device dtype shape -> Parameter Source #

data ToDependent Source #

Constructors

ToDependent 

Instances

Instances details
Apply' ToDependent (Parameter device dtype shape) (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

apply' :: ToDependent -> Parameter device dtype shape -> Tensor device dtype shape Source #

data MakeIndependent Source #

Constructors

MakeIndependent 

Instances

Instances details
Apply' MakeIndependent (Tensor device dtype shape) (IO (Parameter device dtype shape)) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

apply' :: MakeIndependent -> Tensor device dtype shape -> IO (Parameter device dtype shape) Source #

type family Parameters f :: [Type] Source #

Instances

Instances details
type Parameters Dropout Source # 
Instance details

Defined in Torch.Typed.NN.Dropout

type Parameters Dropout = GParameters (Rep Dropout)
type Parameters GD Source # 
Instance details

Defined in Torch.Typed.Optim

type Parameters GD = '[] :: [Type]
type Parameters Double Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters Double = '[] :: [Type]
type Parameters Float Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters Float = '[] :: [Type]
type Parameters Int Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters Int = '[] :: [Type]
type Parameters (Adam momenta) Source # 
Instance details

Defined in Torch.Typed.Optim

type Parameters (Adam momenta) = AdamIter ': (momenta ++ momenta)
type Parameters (GDM momenta) Source # 
Instance details

Defined in Torch.Typed.Optim

type Parameters (GDM momenta) = momenta
type Parameters (HList (f ': fs)) Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (HList (f ': fs)) = Parameters f ++ Parameters (HList fs)
type Parameters (HList ('[] :: [k])) Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (HList ('[] :: [k])) = '[] :: [Type]
type Parameters (LayerNorm normalizedShape dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Normalization

type Parameters (LayerNorm normalizedShape dtype device) = GParameters (Rep (LayerNorm normalizedShape dtype device))
type Parameters (Parameter device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (Parameter device dtype shape) = '[Parameter device dtype shape]
type Parameters (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.Parameter

type Parameters (Tensor device dtype shape) = '[] :: [Type]
type Parameters (Linear inputFeatures outputFeatures dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Linear

type Parameters (Linear inputFeatures outputFeatures dtype device) = GParameters (Rep (Linear inputFeatures outputFeatures dtype device))
type Parameters (GRUCell inputDim hiddenDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

type Parameters (GRUCell inputDim hiddenDim dtype device) = GParameters (Rep (GRUCell inputDim hiddenDim dtype device))
type Parameters (LSTMCell inputDim hiddenDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.LSTM

type Parameters (LSTMCell inputDim hiddenDim dtype device) = GParameters (Rep (LSTMCell inputDim hiddenDim dtype device))
type Parameters (TransformerMLP embedDim ffnDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (TransformerMLP embedDim ffnDim dtype device) = GParameters (Rep (TransformerMLP embedDim ffnDim dtype device))
type Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) = GParameters (Rep (Conv1d inputChannelSize outputChannelSize kernelSize dtype device))
type Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) = GParameters (Rep (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device))
type Parameters (GRULayer inputSize hiddenSize 'Bidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRULayer inputSize hiddenSize 'Bidirectional dtype device) = '[Parameter device dtype (GRUWIShape hiddenSize inputSize), Parameter device dtype (GRUWHShape hiddenSize inputSize), Parameter device dtype (GRUBIShape hiddenSize inputSize), Parameter device dtype (GRUBHShape hiddenSize inputSize), Parameter device dtype (GRUWIShape hiddenSize inputSize), Parameter device dtype (GRUWHShape hiddenSize inputSize), Parameter device dtype (GRUBIShape hiddenSize inputSize), Parameter device dtype (GRUBHShape hiddenSize inputSize)]
type Parameters (GRULayer inputSize hiddenSize 'Unidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRULayer inputSize hiddenSize 'Unidirectional dtype device) = '[Parameter device dtype (GRUWIShape hiddenSize inputSize), Parameter device dtype (GRUWHShape hiddenSize inputSize), Parameter device dtype (GRUBIShape hiddenSize inputSize), Parameter device dtype (GRUBHShape hiddenSize inputSize)]
type Parameters (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) = '[Parameter device dtype (LSTMWIShape hiddenSize inputSize), Parameter device dtype (LSTMWHShape hiddenSize inputSize), Parameter device dtype (LSTMBIShape hiddenSize inputSize), Parameter device dtype (LSTMBHShape hiddenSize inputSize), Parameter device dtype (LSTMWIShape hiddenSize inputSize), Parameter device dtype (LSTMWHShape hiddenSize inputSize), Parameter device dtype (LSTMBIShape hiddenSize inputSize), Parameter device dtype (LSTMBHShape hiddenSize inputSize)]
type Parameters (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) = '[Parameter device dtype (LSTMWIShape hiddenSize inputSize), Parameter device dtype (LSTMWHShape hiddenSize inputSize), Parameter device dtype (LSTMBIShape hiddenSize inputSize), Parameter device dtype (LSTMBHShape hiddenSize inputSize)]
type Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) = GParameters (Rep (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device))
type Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) = GParameters (Rep (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device))
type Parameters (GRU inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRU inputSize hiddenSize numLayers directionality dtype device) = GParameters (Rep (GRU inputSize hiddenSize numLayers directionality dtype device))
type Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) = GRULayerStackParameters (2 <=? numLayers) inputSize hiddenSize numLayers directionality dtype device
type Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTM inputSize hiddenSize numLayers directionality dtype device) = GParameters (Rep (LSTM inputSize hiddenSize numLayers directionality dtype device))
type Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) = LSTMLayerStackParameters (2 <=? numLayers) inputSize hiddenSize numLayers directionality dtype device
type Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) = GParameters (Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device))
type Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) = GParameters (Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device))
type Parameters (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) = GParameters (Rep (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device))
type Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) = GParameters (Rep (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device))
type Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

type Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) = GParameters (Rep (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device))
type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) = GParameters (Rep (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device))
type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

type Parameters (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) = GParameters (Rep (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device))
type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) = GParameters (Rep (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device))
type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

type Parameters (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) = GParameters (Rep (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device))
type Parameters (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) = GParameters (Rep (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device))
type Parameters (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

type Parameters (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) = GParameters (Rep (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device))

type family GParameters (f :: Type -> Type) :: [Type] Source #

Instances

Instances details
type GParameters (U1 :: Type -> Type) Source # 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (U1 :: Type -> Type) = '[] :: [Type]
type GParameters (l :*: r) Source # 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (l :*: r) = GParameters l ++ GParameters r
type GParameters (K1 i f :: Type -> Type) Source # 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (K1 i f :: Type -> Type) = Parameters f
type GParameters (M1 i t f) Source # 
Instance details

Defined in Torch.Typed.Parameter

type GParameters (M1 i t f) = GParameters f

class Randomizable spec f | spec -> f where Source #

Methods

sample :: spec -> IO f Source #

Instances

Instances details
Randomizable BatchNormSpec BatchNorm Source # 
Instance details

Defined in Torch.NN

Methods

sample :: BatchNormSpec -> IO BatchNorm Source #

Randomizable Conv1dSpec Conv1d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv1dSpec -> IO Conv1d Source #

Randomizable Conv2dSpec Conv2d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv2dSpec -> IO Conv2d Source #

Randomizable Conv3dSpec Conv3d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv3dSpec -> IO Conv3d Source #

Randomizable ConvTranspose1dSpec ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose1dSpec -> IO ConvTranspose1d Source #

Randomizable ConvTranspose2dSpec ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose2dSpec -> IO ConvTranspose2d Source #

Randomizable ConvTranspose3dSpec ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose3dSpec -> IO ConvTranspose3d Source #

Randomizable InstanceNormSpec InstanceNorm Source # 
Instance details

Defined in Torch.NN

Methods

sample :: InstanceNormSpec -> IO InstanceNorm Source #

Randomizable LinearSpec Linear Source # 
Instance details

Defined in Torch.NN

Methods

sample :: LinearSpec -> IO Linear Source #

Randomizable UpSampleSpec UpSample Source # 
Instance details

Defined in Torch.NN

Methods

sample :: UpSampleSpec -> IO UpSample Source #

Randomizable ElmanSpec ElmanCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.Elman

Methods

sample :: ElmanSpec -> IO ElmanCell Source #

Randomizable GRUSpec GRUCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

sample :: GRUSpec -> IO GRUCell Source #

Randomizable LSTMSpec LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

sample :: LSTMSpec -> IO LSTMCell Source #

Randomizable DropoutSpec Dropout Source # 
Instance details

Defined in Torch.Typed.NN.Dropout

Methods

sample :: DropoutSpec -> IO Dropout Source #

(Randomizable xSpec x, Randomizable (HList xsSpec) (HList xs)) => Randomizable (HList (xSpec ': xsSpec)) (HList (x ': xs)) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

sample :: HList (xSpec ': xsSpec) -> IO (HList (x ': xs)) Source #

Randomizable (HList ('[] :: [Type])) (HList ('[] :: [Type])) Source # 
Instance details

Defined in Torch.Typed.Parameter

Methods

sample :: HList ('[] :: [Type]) -> IO (HList ('[] :: [Type])) Source #

(TensorOptions normalizedShape dtype device, RandDTypeIsValid device dtype) => Randomizable (LayerNormSpec normalizedShape dtype device) (LayerNorm normalizedShape dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Normalization

Methods

sample :: LayerNormSpec normalizedShape dtype device -> IO (LayerNorm normalizedShape dtype device) Source #

(KnownNat inputFeatures, KnownNat outputFeatures, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (LinearSpec inputFeatures outputFeatures dtype device) (Linear inputFeatures outputFeatures dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Linear

Methods

sample :: LinearSpec inputFeatures outputFeatures dtype device -> IO (Linear inputFeatures outputFeatures dtype device) Source #

(KnownDevice device, KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, RandDTypeIsValid device dtype) => Randomizable (GRUCellSpec inputDim hiddenDim dtype device) (GRUCell inputDim hiddenDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

sample :: GRUCellSpec inputDim hiddenDim dtype device -> IO (GRUCell inputDim hiddenDim dtype device) Source #

(KnownDevice device, KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, RandDTypeIsValid device dtype) => Randomizable (LSTMCellSpec inputDim hiddenDim dtype device) (LSTMCell inputDim hiddenDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.LSTM

Methods

sample :: LSTMCellSpec inputDim hiddenDim dtype device -> IO (LSTMCell inputDim hiddenDim dtype device) Source #

(All KnownNat '[embedDim, ffnDim], KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (TransformerMLPSpec embedDim ffnDim dtype device) (TransformerMLP embedDim ffnDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Methods

sample :: TransformerMLPSpec embedDim ffnDim dtype device -> IO (TransformerMLP embedDim ffnDim dtype device) Source #

(KnownNat inputChannelSize, KnownNat outputChannelSize, KnownNat kernelSize, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device) (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

sample :: Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> IO (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) Source #

(KnownNat inputChannelSize, KnownNat outputChannelSize, KnownNat kernelSize, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device) (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

sample :: ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> IO (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) Source #

(RandDTypeIsValid device dtype, KnownNat inputSize, KnownNat hiddenSize, KnownDType dtype, KnownDevice device) => Randomizable (GRULayerSpec inputSize hiddenSize 'Bidirectional dtype device) (GRULayer inputSize hiddenSize 'Bidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Methods

sample :: GRULayerSpec inputSize hiddenSize 'Bidirectional dtype device -> IO (GRULayer inputSize hiddenSize 'Bidirectional dtype device) Source #

(RandDTypeIsValid device dtype, KnownNat inputSize, KnownNat hiddenSize, KnownDType dtype, KnownDevice device) => Randomizable (GRULayerSpec inputSize hiddenSize 'Unidirectional dtype device) (GRULayer inputSize hiddenSize 'Unidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Methods

sample :: GRULayerSpec inputSize hiddenSize 'Unidirectional dtype device -> IO (GRULayer inputSize hiddenSize 'Unidirectional dtype device) Source #

(RandDTypeIsValid device dtype, KnownNat inputSize, KnownNat hiddenSize, KnownDType dtype, KnownDevice device) => Randomizable (LSTMLayerSpec inputSize hiddenSize 'Bidirectional dtype device) (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

sample :: LSTMLayerSpec inputSize hiddenSize 'Bidirectional dtype device -> IO (LSTMLayer inputSize hiddenSize 'Bidirectional dtype device) Source #

(RandDTypeIsValid device dtype, KnownNat inputSize, KnownNat hiddenSize, KnownDType dtype, KnownDevice device) => Randomizable (LSTMLayerSpec inputSize hiddenSize 'Unidirectional dtype device) (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

sample :: LSTMLayerSpec inputSize hiddenSize 'Unidirectional dtype device -> IO (LSTMLayer inputSize hiddenSize 'Unidirectional dtype device) Source #

(KnownNat inputChannelSize, KnownNat outputChannelSize, KnownNat kernelSize0, KnownNat kernelSize1, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

sample :: Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> IO (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source #

(KnownNat inputChannelSize, KnownNat outputChannelSize, KnownNat kernelSize0, KnownNat kernelSize1, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

sample :: ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> IO (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source #

(1 <= numLayers, (2 <=? numLayers) ~ flag, RandDTypeIsValid device dtype, KnownDType dtype, KnownDevice device, GRULayerStackRandomizable flag inputSize hiddenSize numLayers directionality dtype device) => Randomizable (GRULayerStackSpec inputSize hiddenSize numLayers directionality dtype device) (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Methods

sample :: GRULayerStackSpec inputSize hiddenSize numLayers directionality dtype device -> IO (GRULayerStack inputSize hiddenSize numLayers directionality dtype device) Source #

(KnownDType dtype, KnownDevice device, KnownNat inputSize, KnownNat hiddenSize, KnownNat (NumberOfDirections directionality), RandDTypeIsValid device dtype, Randomizable (GRULayerStackSpec inputSize hiddenSize numLayers directionality dtype device) (GRULayerStack inputSize hiddenSize numLayers directionality dtype device), 1 <= numLayers) => Randomizable (GRUSpec inputSize hiddenSize numLayers directionality dtype device) (GRU inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Methods

sample :: GRUSpec inputSize hiddenSize numLayers directionality dtype device -> IO (GRU inputSize hiddenSize numLayers directionality dtype device) Source #

(1 <= numLayers, (2 <=? numLayers) ~ flag, RandDTypeIsValid device dtype, KnownDType dtype, KnownDevice device, LSTMLayerStackRandomizable flag inputSize hiddenSize numLayers directionality dtype device) => Randomizable (LSTMLayerStackSpec inputSize hiddenSize numLayers directionality dtype device) (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

sample :: LSTMLayerStackSpec inputSize hiddenSize numLayers directionality dtype device -> IO (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) Source #

(KnownDType dtype, KnownDevice device, KnownNat inputSize, KnownNat hiddenSize, KnownNat (NumberOfDirections directionality), RandDTypeIsValid device dtype, Randomizable (LSTMLayerStackSpec inputSize hiddenSize numLayers directionality dtype device) (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device), 1 <= numLayers) => Randomizable (LSTMSpec inputSize hiddenSize numLayers directionality dtype device) (LSTM inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

sample :: LSTMSpec inputSize hiddenSize numLayers directionality dtype device -> IO (LSTM inputSize hiddenSize numLayers directionality dtype device) Source #

(paddingIdx <= numEmbeds, 1 <= (numEmbeds - paddingIdx), (((numEmbeds - paddingIdx) - 1) + (1 + paddingIdx)) ~ numEmbeds, KnownNat paddingIdx, KnownNat numEmbeds, KnownNat embedSize, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (EmbeddingSpec ('Just paddingIdx) numEmbeds embedSize 'Learned dtype device) (Embedding ('Just paddingIdx) numEmbeds embedSize 'Learned dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

Methods

sample :: EmbeddingSpec ('Just paddingIdx) numEmbeds embedSize 'Learned dtype device -> IO (Embedding ('Just paddingIdx) numEmbeds embedSize 'Learned dtype device) Source #

(KnownNat numEmbeds, KnownNat embedSize, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (EmbeddingSpec ('Nothing :: Maybe Nat) numEmbeds embedSize 'Learned dtype device) (Embedding ('Nothing :: Maybe Nat) numEmbeds embedSize 'Learned dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

Methods

sample :: EmbeddingSpec ('Nothing :: Maybe Nat) numEmbeds embedSize 'Learned dtype device -> IO (Embedding ('Nothing :: Maybe Nat) numEmbeds embedSize 'Learned dtype device) Source #

Randomizable (EmbeddingSpec paddingIdx numEmbeds embedSize 'Constant dtype device) (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

Methods

sample :: EmbeddingSpec paddingIdx numEmbeds embedSize 'Constant dtype device -> IO (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) Source #

(All KnownNat '[embedDim, kEmbedDim, vEmbedDim, numHeads], KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (MultiheadAttentionSpec embedDim kEmbedDim vEmbedDim numHeads dtype device) (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Methods

sample :: MultiheadAttentionSpec embedDim kEmbedDim vEmbedDim numHeads dtype device -> IO (MultiheadAttention embedDim kEmbedDim vEmbedDim numHeads dtype device) Source #

(KnownNat inputChannelSize, KnownNat outputChannelSize, KnownNat kernelSize0, KnownNat kernelSize1, KnownNat kernelSize2, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

sample :: Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> IO (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source #

(KnownNat inputChannelSize, KnownNat outputChannelSize, KnownNat kernelSize0, KnownNat kernelSize1, KnownNat kernelSize2, KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

sample :: ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> IO (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source #

(KnownNat hiddenSize, KnownNat numLayers, KnownNat (NumberOfDirections directionality), KnownDType dtype, KnownDevice device, Randomizable (GRUSpec inputSize hiddenSize numLayers directionality dtype device) (GRU inputSize hiddenSize numLayers directionality dtype device)) => Randomizable (GRUWithInitSpec inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Methods

sample :: GRUWithInitSpec inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device -> IO (GRUWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source #

(KnownNat hiddenSize, KnownNat numLayers, KnownNat (NumberOfDirections directionality), KnownDType dtype, KnownDevice device, Randomizable (GRUSpec inputSize hiddenSize numLayers directionality dtype device) (GRU inputSize hiddenSize numLayers directionality dtype device)) => Randomizable (GRUWithInitSpec inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.GRU

Methods

sample :: GRUWithInitSpec inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device -> IO (GRUWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source #

(KnownNat hiddenSize, KnownNat numLayers, KnownNat (NumberOfDirections directionality), KnownDType dtype, KnownDevice device, Randomizable (LSTMSpec inputSize hiddenSize numLayers directionality dtype device) (LSTM inputSize hiddenSize numLayers directionality dtype device)) => Randomizable (LSTMWithInitSpec inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

sample :: LSTMWithInitSpec inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device -> IO (LSTMWithInit inputSize hiddenSize numLayers directionality 'ConstantInitialization dtype device) Source #

(KnownNat hiddenSize, KnownNat numLayers, KnownNat (NumberOfDirections directionality), KnownDType dtype, KnownDevice device, Randomizable (LSTMSpec inputSize hiddenSize numLayers directionality dtype device) (LSTM inputSize hiddenSize numLayers directionality dtype device)) => Randomizable (LSTMWithInitSpec inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

sample :: LSTMWithInitSpec inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device -> IO (LSTMWithInit inputSize hiddenSize numLayers directionality 'LearnedInitialization dtype device) Source #

(All KnownNat '[embedDim, kEmbedDim, vEmbedDim, numHeads, ffnDim], KnownDType dtype, KnownDevice device, RandDTypeIsValid device dtype) => Randomizable (TransformerLayerSpec embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Methods

sample :: TransformerLayerSpec embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device -> IO (TransformerLayer embedDim kEmbedDim vEmbedDim numHeads ffnDim dtype device) Source #

(paddingIdx <= numEmbeds, 1 <= (numEmbeds - paddingIdx), 1 <= Div embedDim 2, (((numEmbeds - paddingIdx) - 1) + (1 + paddingIdx)) ~ numEmbeds, (Div embedDim 2 * 2) ~ embedDim, All KnownNat '[ffnDim, paddingIdx, numEmbeds, embedDim], HReplicate numAttnLayers (TransformerLayerSpec embedDim embedDim embedDim numHeads ffnDim dtype device), Randomizable (HList (HReplicateR numAttnLayers (TransformerLayerSpec embedDim embedDim embedDim numHeads ffnDim dtype device))) (HList (HReplicateR numAttnLayers (TransformerLayer embedDim embedDim embedDim numHeads ffnDim dtype device))), KnownDType dtype, RandDTypeIsValid device dtype, StandardFloatingPointDTypeValidation device 'Float, BasicArithmeticDTypeIsValid device 'Float, KnownDevice device) => Randomizable (TransformerLMSpec numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Methods

sample :: TransformerLMSpec numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device -> IO (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) Source #

Orphan instances

(Randomizable xSpec x, Randomizable (HList xsSpec) (HList xs)) => Randomizable (HList (xSpec ': xsSpec)) (HList (x ': xs)) Source # 
Instance details

Methods

sample :: HList (xSpec ': xsSpec) -> IO (HList (x ': xs)) Source #

Randomizable (HList ('[] :: [Type])) (HList ('[] :: [Type])) Source # 
Instance details

Methods

sample :: HList ('[] :: [Type]) -> IO (HList ('[] :: [Type])) Source #