hasktorch-0.2.0.0: Functional differentiable programming in Haskell
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.NN

Documentation

type Parameter = IndependentTensor Source #

type ParamStream a = State [Parameter] a Source #

nextParameter :: ParamStream Parameter Source #

class HasForward f a b | f a -> b where Source #

Minimal complete definition

Nothing

Methods

forward :: f -> a -> b Source #

default forward :: (Generic f, Generic a, Generic b, GHasForward (Rep f) (Rep a) (Rep b)) => f -> a -> b Source #

forwardStoch :: f -> a -> IO b Source #

default forwardStoch :: (Generic f, Generic a, Generic b, GHasForward (Rep f) (Rep a) (Rep b)) => f -> a -> IO b Source #

Instances

Instances details
HasForward Linear Tensor Tensor Source # 
Instance details

Defined in Torch.NN

Methods

forward :: Linear -> Tensor -> Tensor Source #

forwardStoch :: Linear -> Tensor -> IO Tensor Source #

HasForward UpSample Tensor Tensor Source # 
Instance details

Defined in Torch.NN

Methods

forward :: UpSample -> Tensor -> Tensor Source #

forwardStoch :: UpSample -> Tensor -> IO Tensor Source #

HasForward ScriptModule [IValue] IValue Source # 
Instance details

Defined in Torch.Script

Methods

forward :: ScriptModule -> [IValue] -> IValue Source #

forwardStoch :: ScriptModule -> [IValue] -> IO IValue Source #

HasForward Dropout (Tensor device dtype shape) (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.NN.Dropout

Methods

forward :: Dropout -> Tensor device dtype shape -> Tensor device dtype shape Source #

forwardStoch :: Dropout -> Tensor device dtype shape -> IO (Tensor device dtype shape) Source #

(IsSuffixOf normalizedShape shape, KnownShape normalizedShape) => HasForward (LayerNorm normalizedShape dtype device) (Tensor device dtype shape) (Tensor device dtype shape) Source # 
Instance details

Defined in Torch.Typed.NN.Normalization

Methods

forward :: LayerNorm normalizedShape dtype device -> Tensor device dtype shape -> Tensor device dtype shape Source #

forwardStoch :: LayerNorm normalizedShape dtype device -> Tensor device dtype shape -> IO (Tensor device dtype shape) Source #

(shape'' ~ MatMul shape '[inputFeatures, outputFeatures], shape' ~ Broadcast shape'' shape'') => HasForward (Linear inputFeatures outputFeatures dtype device) (Tensor device dtype shape) (Tensor device dtype shape') Source # 
Instance details

Defined in Torch.Typed.NN.Linear

Methods

forward :: Linear inputFeatures outputFeatures dtype device -> Tensor device dtype shape -> Tensor device dtype shape' Source #

forwardStoch :: Linear inputFeatures outputFeatures dtype device -> Tensor device dtype shape -> IO (Tensor device dtype shape') Source #

(All KnownNat '[stride, padding, inputChannelSize, outputChannelSize, kernelSize, inputSize, batchSize, outputSize], ConvSideCheck inputSize kernelSize stride padding outputSize) => HasForward (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) (Tensor device dtype '[batchSize, inputChannelSize, inputSize], Proxy stride, Proxy padding) (Tensor device dtype '[batchSize, outputChannelSize, outputSize]) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

forward :: Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize], Proxy stride, Proxy padding) -> Tensor device dtype '[batchSize, outputChannelSize, outputSize] Source #

forwardStoch :: Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize], Proxy stride, Proxy padding) -> IO (Tensor device dtype '[batchSize, outputChannelSize, outputSize]) Source #

(All KnownNat '[stride, padding, inputChannelSize, outputChannelSize, kernelSize, inputSize, batchSize, outputSize], ConvSideCheck inputSize kernelSize stride padding outputSize) => HasForward (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) (Tensor device dtype '[batchSize, inputChannelSize, inputSize], Proxy stride, Proxy padding) (Tensor device dtype '[batchSize, outputChannelSize, outputSize]) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

forward :: ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize], Proxy stride, Proxy padding) -> Tensor device dtype '[batchSize, outputChannelSize, outputSize] Source #

forwardStoch :: ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize], Proxy stride, Proxy padding) -> IO (Tensor device dtype '[batchSize, outputChannelSize, outputSize]) Source #

(All KnownNat '[Fst stride, Snd stride, Fst padding, Snd padding, inputChannelSize, outputChannelSize, kernelSize0, kernelSize1, inputSize0, inputSize1, batchSize, outputSize0, outputSize1], ConvSideCheck inputSize0 kernelSize0 (Fst stride) (Fst padding) outputSize0, ConvSideCheck inputSize1 kernelSize1 (Snd stride) (Snd padding) outputSize1) => HasForward (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1], Proxy stride, Proxy padding) (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1]) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

forward :: Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1], Proxy stride, Proxy padding) -> Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1] Source #

forwardStoch :: Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1], Proxy stride, Proxy padding) -> IO (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1]) Source #

(All KnownNat '[Fst stride, Snd stride, Fst padding, Snd padding, inputChannelSize, outputChannelSize, kernelSize0, kernelSize1, inputSize0, inputSize1, batchSize, outputSize0, outputSize1], ConvSideCheck inputSize0 kernelSize0 (Fst stride) (Fst padding) outputSize0, ConvSideCheck inputSize1 kernelSize1 (Snd stride) (Snd padding) outputSize1) => HasForward (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1], Proxy stride, Proxy padding) (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1]) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

forward :: ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1], Proxy stride, Proxy padding) -> Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1] Source #

forwardStoch :: ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1], Proxy stride, Proxy padding) -> IO (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1]) Source #

(KnownMaybeNat paddingIdx, PaddingIdxCheck paddingIdx numEmbeds, shape' ~ Reverse (embedSize ': Reverse shape)) => HasForward (Embedding paddingIdx numEmbeds embedSize embeddingType dtype device) (Tensor device 'Int64 shape) (Tensor device dtype shape') Source # 
Instance details

Defined in Torch.Typed.NN.Sparse

Methods

forward :: Embedding paddingIdx numEmbeds embedSize embeddingType dtype device -> Tensor device 'Int64 shape -> Tensor device dtype shape' Source #

forwardStoch :: Embedding paddingIdx numEmbeds embedSize embeddingType dtype device -> Tensor device 'Int64 shape -> IO (Tensor device dtype shape') Source #

(All KnownNat '[Fst3 stride, Snd3 stride, Trd3 stride, Fst3 padding, Snd3 padding, Trd3 padding, inputChannelSize, outputChannelSize, kernelSize0, kernelSize1, kernelSize2, inputSize0, inputSize1, inputSize2, batchSize], ConvSideCheck inputSize0 kernelSize0 (Fst3 stride) (Fst3 padding) outputSize0, ConvSideCheck inputSize1 kernelSize1 (Snd3 stride) (Snd3 padding) outputSize1, ConvSideCheck inputSize2 kernelSize2 (Trd3 stride) (Trd3 padding) outputSize2) => HasForward (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1, inputSize2], Proxy stride, Proxy padding) (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1, outputSize2]) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

forward :: Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1, inputSize2], Proxy stride, Proxy padding) -> Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1, outputSize2] Source #

forwardStoch :: Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1, inputSize2], Proxy stride, Proxy padding) -> IO (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1, outputSize2]) Source #

(All KnownNat '[Fst3 stride, Snd3 stride, Trd3 stride, Fst3 padding, Snd3 padding, Trd3 padding, inputChannelSize, outputChannelSize, kernelSize0, kernelSize1, kernelSize2, inputSize0, inputSize1, inputSize2, batchSize], ConvSideCheck inputSize0 kernelSize0 (Fst3 stride) (Fst3 padding) outputSize0, ConvSideCheck inputSize1 kernelSize1 (Snd3 stride) (Snd3 padding) outputSize1, ConvSideCheck inputSize2 kernelSize2 (Trd3 stride) (Trd3 padding) outputSize2) => HasForward (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1, inputSize2], Proxy stride, Proxy padding) (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1, outputSize2]) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

forward :: ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1, inputSize2], Proxy stride, Proxy padding) -> Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1, outputSize2] Source #

forwardStoch :: ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> (Tensor device dtype '[batchSize, inputChannelSize, inputSize0, inputSize1, inputSize2], Proxy stride, Proxy padding) -> IO (Tensor device dtype '[batchSize, outputChannelSize, outputSize0, outputSize1, outputSize2]) Source #

(All KnownNat '[paddingIdx, embedDim, seqLen, batchSize], (paddingIdx + 1) <= numEmbeds, 1 <= seqLen, HFoldrM IO (FoldLayers batchSize seqLen dtype device) (Tensor device dtype '[batchSize, seqLen, embedDim]) (HReplicateR numAttnLayers (TransformerLayer embedDim embedDim embedDim numHeads ffnDim dtype device)) (Tensor device dtype '[batchSize, seqLen, embedDim]), BasicArithmeticDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device 'Int64, KnownDType dtype, KnownDevice device) => HasForward (TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device) (Tensor device 'Int64 '[batchSize, seqLen]) (Tensor device dtype '[batchSize, seqLen, numEmbeds]) Source # 
Instance details

Defined in Torch.Typed.NN.Transformer

Methods

forward :: TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device -> Tensor device 'Int64 '[batchSize, seqLen] -> Tensor device dtype '[batchSize, seqLen, numEmbeds] Source #

forwardStoch :: TransformerLM numAttnLayers numHeads ffnDim paddingIdx numEmbeds embedDim dtype device -> Tensor device 'Int64 '[batchSize, seqLen] -> IO (Tensor device dtype '[batchSize, seqLen, numEmbeds]) Source #

class GHasForward (f :: Type -> Type) (a :: Type -> Type) (b :: Type -> Type) | f a -> b where Source #

Methods

gForward :: forall c c' c''. f c -> a c' -> b c'' Source #

gForwardStoch :: forall c c' c''. f c -> a c' -> IO (b c) Source #

Instances

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

Defined in Torch.NN

Methods

gForward :: U1 c -> U1 c' -> U1 c'' Source #

gForwardStoch :: forall c c' c''. U1 c -> U1 c' -> IO (U1 c) Source #

(GHasForward f a b, GHasForward g a' b', b'' ~ (b :*: b')) => GHasForward (f :*: g) (a :*: a') b'' Source # 
Instance details

Defined in Torch.NN

Methods

gForward :: (f :*: g) c -> (a :*: a') c' -> b'' c'' Source #

gForwardStoch :: forall c c' c''. (f :*: g) c -> (a :*: a') c' -> IO (b'' c) Source #

(GHasForward f a b, GHasForward g a' b', b'' ~ (b :+: b')) => GHasForward (f :+: g) (a :+: a') b'' Source # 
Instance details

Defined in Torch.NN

Methods

gForward :: (f :+: g) c -> (a :+: a') c' -> b'' c'' Source #

gForwardStoch :: forall c c' c''. (f :+: g) c -> (a :+: a') c' -> IO (b'' c) Source #

HasForward f a b => GHasForward (K1 i f :: Type -> Type) (K1 i a :: Type -> Type) (K1 i b :: Type -> Type) Source # 
Instance details

Defined in Torch.NN

Methods

gForward :: K1 i f c -> K1 i a c' -> K1 i b c'' Source #

gForwardStoch :: forall c c' c''. K1 i f c -> K1 i a c' -> IO (K1 i b c) Source #

GHasForward f a b => GHasForward (M1 i t f) (M1 i t' a) (M1 i t' b) Source # 
Instance details

Defined in Torch.NN

Methods

gForward :: M1 i t f c -> M1 i t' a c' -> M1 i t' b c'' Source #

gForwardStoch :: forall c c' c''. M1 i t f c -> M1 i t' a c' -> IO (M1 i t' b c) Source #

class Parameterized f where Source #

Minimal complete definition

Nothing

Methods

flattenParameters :: f -> [Parameter] Source #

default flattenParameters :: (Generic f, GParameterized (Rep f)) => f -> [Parameter] Source #

_replaceParameters :: f -> ParamStream f Source #

default _replaceParameters :: (Generic f, GParameterized (Rep f)) => f -> ParamStream f Source #

Instances

Instances details
Parameterized Conv1d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Conv1d -> [Parameter] Source #

_replaceParameters :: Conv1d -> ParamStream Conv1d Source #

Parameterized Conv2d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Conv2d -> [Parameter] Source #

_replaceParameters :: Conv2d -> ParamStream Conv2d Source #

Parameterized Conv3d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Conv3d -> [Parameter] Source #

_replaceParameters :: Conv3d -> ParamStream Conv3d Source #

Parameterized ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: ConvTranspose1d -> [Parameter] Source #

_replaceParameters :: ConvTranspose1d -> ParamStream ConvTranspose1d Source #

Parameterized ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: ConvTranspose2d -> [Parameter] Source #

_replaceParameters :: ConvTranspose2d -> ParamStream ConvTranspose2d Source #

Parameterized ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: ConvTranspose3d -> [Parameter] Source #

_replaceParameters :: ConvTranspose3d -> ParamStream ConvTranspose3d Source #

Parameterized Linear Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Linear -> [Parameter] Source #

_replaceParameters :: Linear -> ParamStream Linear Source #

Parameterized Parameter Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Parameter -> [Parameter] Source #

_replaceParameters :: Parameter -> ParamStream Parameter Source #

Parameterized UpSample Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: UpSample -> [Parameter] Source #

_replaceParameters :: UpSample -> ParamStream UpSample Source #

Parameterized UpSampleSpec Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: UpSampleSpec -> [Parameter] Source #

_replaceParameters :: UpSampleSpec -> ParamStream UpSampleSpec Source #

Parameterized ElmanCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.Elman

Methods

flattenParameters :: ElmanCell -> [Parameter] Source #

_replaceParameters :: ElmanCell -> ParamStream ElmanCell Source #

Parameterized GRUCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

flattenParameters :: GRUCell -> [Parameter] Source #

_replaceParameters :: GRUCell -> ParamStream GRUCell Source #

Parameterized LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

flattenParameters :: LSTMCell -> [Parameter] Source #

_replaceParameters :: LSTMCell -> ParamStream LSTMCell Source #

Parameterized ScriptModule Source # 
Instance details

Defined in Torch.Script

Methods

flattenParameters :: ScriptModule -> [Parameter] Source #

_replaceParameters :: ScriptModule -> ParamStream ScriptModule Source #

Parameterized Tensor Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Tensor -> [Parameter] Source #

_replaceParameters :: Tensor -> ParamStream Tensor Source #

Scalar a => Parameterized a Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: a -> [Parameter] Source #

_replaceParameters :: a -> ParamStream a Source #

(Foldable t, Traversable t, Parameterized a) => Parameterized (t a) Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: t a -> [Parameter] Source #

_replaceParameters :: t a -> ParamStream (t a) Source #

Parameterized (a -> a) Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: (a -> a) -> [Parameter] Source #

_replaceParameters :: (a -> a) -> ParamStream (a -> a) Source #

(Parameterized a, Parameterized b) => Parameterized (a, b) Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: (a, b) -> [Parameter] Source #

_replaceParameters :: (a, b) -> ParamStream (a, b) Source #

(Parameterized a, Parameterized b, Parameterized c) => Parameterized (a, b, c) Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: (a, b, c) -> [Parameter] Source #

_replaceParameters :: (a, b, c) -> ParamStream (a, b, c) Source #

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

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

flattenParameters :: LSTMLayer inputSize hiddenSize directionality dtype device -> [Parameter] Source #

_replaceParameters :: LSTMLayer inputSize hiddenSize directionality dtype device -> ParamStream (LSTMLayer inputSize hiddenSize directionality dtype device) Source #

Parameterized (LSTM inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

flattenParameters :: LSTM inputSize hiddenSize numLayers directionality dtype device -> [Parameter] Source #

_replaceParameters :: LSTM inputSize hiddenSize numLayers directionality dtype device -> ParamStream (LSTM inputSize hiddenSize numLayers directionality dtype device) Source #

Parameterized (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

flattenParameters :: LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device -> [Parameter] Source #

_replaceParameters :: LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device -> ParamStream (LSTMLayerStack inputSize hiddenSize numLayers directionality dtype device) Source #

Parameterized (LSTMWithInit inputSize hiddenSize numLayers directionality initialization dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.LSTM

Methods

flattenParameters :: LSTMWithInit inputSize hiddenSize numLayers directionality initialization dtype device -> [Parameter] Source #

_replaceParameters :: LSTMWithInit inputSize hiddenSize numLayers directionality initialization dtype device -> ParamStream (LSTMWithInit inputSize hiddenSize numLayers directionality initialization dtype device) Source #

replaceParameters :: Parameterized f => f -> [Parameter] -> f Source #

class GParameterized f where Source #

Methods

gFlattenParameters :: forall a. f a -> [Parameter] Source #

_gReplaceParameters :: forall a. f a -> ParamStream (f a) Source #

Instances

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

Defined in Torch.NN

Methods

gFlattenParameters :: U1 a -> [Parameter] Source #

_gReplaceParameters :: U1 a -> ParamStream (U1 a) Source #

(GParameterized f, GParameterized g) => GParameterized (f :*: g) Source # 
Instance details

Defined in Torch.NN

Methods

gFlattenParameters :: (f :*: g) a -> [Parameter] Source #

_gReplaceParameters :: (f :*: g) a -> ParamStream ((f :*: g) a) Source #

(GParameterized f, GParameterized g) => GParameterized (f :+: g) Source # 
Instance details

Defined in Torch.NN

Methods

gFlattenParameters :: (f :+: g) a -> [Parameter] Source #

_gReplaceParameters :: (f :+: g) a -> ParamStream ((f :+: g) a) Source #

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

Defined in Torch.NN

Methods

gFlattenParameters :: K1 i c a -> [Parameter] Source #

_gReplaceParameters :: K1 i c a -> ParamStream (K1 i c a) Source #

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

Defined in Torch.NN

Methods

gFlattenParameters :: M1 i t f a -> [Parameter] Source #

_gReplaceParameters :: M1 i t f a -> ParamStream (M1 i t f a) Source #

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 '[] -> IO (HList '[]) 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 numEmbeds embedSize 'Learned dtype device -> IO (Embedding 'Nothing 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 #

data LinearSpec Source #

Constructors

LinearSpec 

Fields

Instances

Instances details
Show LinearSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> LinearSpec -> ShowS Source #

show :: LinearSpec -> String Source #

showList :: [LinearSpec] -> ShowS Source #

Eq LinearSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: LinearSpec -> LinearSpec -> Bool Source #

(/=) :: LinearSpec -> LinearSpec -> Bool Source #

Randomizable LinearSpec Linear Source # 
Instance details

Defined in Torch.NN

Methods

sample :: LinearSpec -> IO Linear Source #

data Linear Source #

Constructors

Linear 

Fields

Instances

Instances details
Generic Linear Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep Linear :: Type -> Type Source #

Methods

from :: Linear -> Rep Linear x Source #

to :: Rep Linear x -> Linear Source #

Show Linear Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Linear -> ShowS Source #

show :: Linear -> String Source #

showList :: [Linear] -> ShowS Source #

Parameterized Linear Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Linear -> [Parameter] Source #

_replaceParameters :: Linear -> ParamStream Linear Source #

Randomizable LinearSpec Linear Source # 
Instance details

Defined in Torch.NN

Methods

sample :: LinearSpec -> IO Linear Source #

HasForward Linear Tensor Tensor Source # 
Instance details

Defined in Torch.NN

Methods

forward :: Linear -> Tensor -> Tensor Source #

forwardStoch :: Linear -> Tensor -> IO Tensor Source #

type Rep Linear Source # 
Instance details

Defined in Torch.NN

type Rep Linear = D1 ('MetaData "Linear" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Linear" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

linear :: Linear -> Tensor -> Tensor Source #

linearForward :: Linear -> Tensor -> Tensor Source #

data Conv1dSpec Source #

Constructors

Conv1dSpec 

Fields

Instances

Instances details
Show Conv1dSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Conv1dSpec -> ShowS Source #

show :: Conv1dSpec -> String Source #

showList :: [Conv1dSpec] -> ShowS Source #

Eq Conv1dSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: Conv1dSpec -> Conv1dSpec -> Bool Source #

(/=) :: Conv1dSpec -> Conv1dSpec -> Bool Source #

Randomizable Conv1dSpec Conv1d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv1dSpec -> IO Conv1d Source #

data Conv1d Source #

Constructors

Conv1d 

Fields

Instances

Instances details
Generic Conv1d Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep Conv1d :: Type -> Type Source #

Methods

from :: Conv1d -> Rep Conv1d x Source #

to :: Rep Conv1d x -> Conv1d Source #

Show Conv1d Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Conv1d -> ShowS Source #

show :: Conv1d -> String Source #

showList :: [Conv1d] -> ShowS Source #

Parameterized Conv1d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Conv1d -> [Parameter] Source #

_replaceParameters :: Conv1d -> ParamStream Conv1d Source #

Randomizable Conv1dSpec Conv1d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv1dSpec -> IO Conv1d Source #

type Rep Conv1d Source # 
Instance details

Defined in Torch.NN

type Rep Conv1d = D1 ('MetaData "Conv1d" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Conv1d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

conv1dForward Source #

Arguments

:: Conv1d

layer

-> Int

stride

-> Int

padding

-> Tensor

input

-> Tensor

output

data Conv2dSpec Source #

Constructors

Conv2dSpec 

Fields

Instances

Instances details
Show Conv2dSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Conv2dSpec -> ShowS Source #

show :: Conv2dSpec -> String Source #

showList :: [Conv2dSpec] -> ShowS Source #

Eq Conv2dSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: Conv2dSpec -> Conv2dSpec -> Bool Source #

(/=) :: Conv2dSpec -> Conv2dSpec -> Bool Source #

Randomizable Conv2dSpec Conv2d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv2dSpec -> IO Conv2d Source #

data Conv2d Source #

Constructors

Conv2d 

Fields

Instances

Instances details
Generic Conv2d Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep Conv2d :: Type -> Type Source #

Methods

from :: Conv2d -> Rep Conv2d x Source #

to :: Rep Conv2d x -> Conv2d Source #

Show Conv2d Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Conv2d -> ShowS Source #

show :: Conv2d -> String Source #

showList :: [Conv2d] -> ShowS Source #

Parameterized Conv2d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Conv2d -> [Parameter] Source #

_replaceParameters :: Conv2d -> ParamStream Conv2d Source #

Randomizable Conv2dSpec Conv2d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv2dSpec -> IO Conv2d Source #

type Rep Conv2d Source # 
Instance details

Defined in Torch.NN

type Rep Conv2d = D1 ('MetaData "Conv2d" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Conv2d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

conv2dForward Source #

Arguments

:: Conv2d

layer

-> (Int, Int)

stride

-> (Int, Int)

padding

-> Tensor

input

-> Tensor

output

data Conv3dSpec Source #

Constructors

Conv3dSpec 

Fields

Instances

Instances details
Show Conv3dSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Conv3dSpec -> ShowS Source #

show :: Conv3dSpec -> String Source #

showList :: [Conv3dSpec] -> ShowS Source #

Eq Conv3dSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: Conv3dSpec -> Conv3dSpec -> Bool Source #

(/=) :: Conv3dSpec -> Conv3dSpec -> Bool Source #

Randomizable Conv3dSpec Conv3d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv3dSpec -> IO Conv3d Source #

data Conv3d Source #

Constructors

Conv3d 

Fields

Instances

Instances details
Generic Conv3d Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep Conv3d :: Type -> Type Source #

Methods

from :: Conv3d -> Rep Conv3d x Source #

to :: Rep Conv3d x -> Conv3d Source #

Show Conv3d Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> Conv3d -> ShowS Source #

show :: Conv3d -> String Source #

showList :: [Conv3d] -> ShowS Source #

Parameterized Conv3d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: Conv3d -> [Parameter] Source #

_replaceParameters :: Conv3d -> ParamStream Conv3d Source #

Randomizable Conv3dSpec Conv3d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: Conv3dSpec -> IO Conv3d Source #

type Rep Conv3d Source # 
Instance details

Defined in Torch.NN

type Rep Conv3d = D1 ('MetaData "Conv3d" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Conv3d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

conv3dForward Source #

Arguments

:: Conv3d

layer

-> (Int, Int, Int)

stride

-> (Int, Int, Int)

padding

-> Tensor

input

-> Tensor

output

data ConvTranspose1dSpec Source #

Constructors

ConvTranspose1dSpec 

Fields

Instances

Instances details
Show ConvTranspose1dSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> ConvTranspose1dSpec -> ShowS Source #

show :: ConvTranspose1dSpec -> String Source #

showList :: [ConvTranspose1dSpec] -> ShowS Source #

Eq ConvTranspose1dSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: ConvTranspose1dSpec -> ConvTranspose1dSpec -> Bool Source #

(/=) :: ConvTranspose1dSpec -> ConvTranspose1dSpec -> Bool Source #

Randomizable ConvTranspose1dSpec ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose1dSpec -> IO ConvTranspose1d Source #

data ConvTranspose1d Source #

Constructors

ConvTranspose1d 

Fields

Instances

Instances details
Generic ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep ConvTranspose1d :: Type -> Type Source #

Methods

from :: ConvTranspose1d -> Rep ConvTranspose1d x Source #

to :: Rep ConvTranspose1d x -> ConvTranspose1d Source #

Show ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> ConvTranspose1d -> ShowS Source #

show :: ConvTranspose1d -> String Source #

showList :: [ConvTranspose1d] -> ShowS Source #

Parameterized ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: ConvTranspose1d -> [Parameter] Source #

_replaceParameters :: ConvTranspose1d -> ParamStream ConvTranspose1d Source #

Randomizable ConvTranspose1dSpec ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose1dSpec -> IO ConvTranspose1d Source #

type Rep ConvTranspose1d Source # 
Instance details

Defined in Torch.NN

type Rep ConvTranspose1d = D1 ('MetaData "ConvTranspose1d" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "ConvTranspose1d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

convTranspose1dForward Source #

Arguments

:: ConvTranspose1d

layer

-> Int

stride

-> Int

padding

-> Tensor

input

-> Tensor

output

data ConvTranspose2dSpec Source #

Constructors

ConvTranspose2dSpec 

Fields

Instances

Instances details
Show ConvTranspose2dSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> ConvTranspose2dSpec -> ShowS Source #

show :: ConvTranspose2dSpec -> String Source #

showList :: [ConvTranspose2dSpec] -> ShowS Source #

Eq ConvTranspose2dSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: ConvTranspose2dSpec -> ConvTranspose2dSpec -> Bool Source #

(/=) :: ConvTranspose2dSpec -> ConvTranspose2dSpec -> Bool Source #

Randomizable ConvTranspose2dSpec ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose2dSpec -> IO ConvTranspose2d Source #

data ConvTranspose2d Source #

Constructors

ConvTranspose2d 

Fields

Instances

Instances details
Generic ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep ConvTranspose2d :: Type -> Type Source #

Methods

from :: ConvTranspose2d -> Rep ConvTranspose2d x Source #

to :: Rep ConvTranspose2d x -> ConvTranspose2d Source #

Show ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> ConvTranspose2d -> ShowS Source #

show :: ConvTranspose2d -> String Source #

showList :: [ConvTranspose2d] -> ShowS Source #

Parameterized ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: ConvTranspose2d -> [Parameter] Source #

_replaceParameters :: ConvTranspose2d -> ParamStream ConvTranspose2d Source #

Randomizable ConvTranspose2dSpec ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose2dSpec -> IO ConvTranspose2d Source #

type Rep ConvTranspose2d Source # 
Instance details

Defined in Torch.NN

type Rep ConvTranspose2d = D1 ('MetaData "ConvTranspose2d" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "ConvTranspose2d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

convTranspose2dForward Source #

Arguments

:: ConvTranspose2d

layer

-> (Int, Int)

stride

-> (Int, Int)

padding

-> Tensor

input

-> Tensor

output

data ConvTranspose3dSpec Source #

Constructors

ConvTranspose3dSpec 

Fields

Instances

Instances details
Show ConvTranspose3dSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> ConvTranspose3dSpec -> ShowS Source #

show :: ConvTranspose3dSpec -> String Source #

showList :: [ConvTranspose3dSpec] -> ShowS Source #

Eq ConvTranspose3dSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: ConvTranspose3dSpec -> ConvTranspose3dSpec -> Bool Source #

(/=) :: ConvTranspose3dSpec -> ConvTranspose3dSpec -> Bool Source #

Randomizable ConvTranspose3dSpec ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose3dSpec -> IO ConvTranspose3d Source #

data ConvTranspose3d Source #

Constructors

ConvTranspose3d 

Fields

Instances

Instances details
Generic ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep ConvTranspose3d :: Type -> Type Source #

Methods

from :: ConvTranspose3d -> Rep ConvTranspose3d x Source #

to :: Rep ConvTranspose3d x -> ConvTranspose3d Source #

Show ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> ConvTranspose3d -> ShowS Source #

show :: ConvTranspose3d -> String Source #

showList :: [ConvTranspose3d] -> ShowS Source #

Parameterized ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: ConvTranspose3d -> [Parameter] Source #

_replaceParameters :: ConvTranspose3d -> ParamStream ConvTranspose3d Source #

Randomizable ConvTranspose3dSpec ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

Methods

sample :: ConvTranspose3dSpec -> IO ConvTranspose3d Source #

type Rep ConvTranspose3d Source # 
Instance details

Defined in Torch.NN

type Rep ConvTranspose3d = D1 ('MetaData "ConvTranspose3d" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "ConvTranspose3d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)))

convTranspose3dForward Source #

Arguments

:: ConvTranspose3d

layer

-> (Int, Int, Int)

stride

-> (Int, Int, Int)

padding

-> Tensor

input

-> Tensor

output

data BatchNormSpec Source #

Constructors

BatchNormSpec 

Fields

Instances

Instances details
Show BatchNormSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> BatchNormSpec -> ShowS Source #

show :: BatchNormSpec -> String Source #

showList :: [BatchNormSpec] -> ShowS Source #

Eq BatchNormSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: BatchNormSpec -> BatchNormSpec -> Bool Source #

(/=) :: BatchNormSpec -> BatchNormSpec -> Bool Source #

Randomizable BatchNormSpec BatchNorm Source # 
Instance details

Defined in Torch.NN

Methods

sample :: BatchNormSpec -> IO BatchNorm Source #

data BatchNorm Source #

Constructors

BatchNorm 

Fields

Instances

Instances details
Generic BatchNorm Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep BatchNorm :: Type -> Type Source #

Methods

from :: BatchNorm -> Rep BatchNorm x Source #

to :: Rep BatchNorm x -> BatchNorm Source #

Show BatchNorm Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> BatchNorm -> ShowS Source #

show :: BatchNorm -> String Source #

showList :: [BatchNorm] -> ShowS Source #

Randomizable BatchNormSpec BatchNorm Source # 
Instance details

Defined in Torch.NN

Methods

sample :: BatchNormSpec -> IO BatchNorm Source #

type Rep BatchNorm Source # 
Instance details

Defined in Torch.NN

type Rep BatchNorm = D1 ('MetaData "BatchNorm" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "BatchNorm" 'PrefixI 'True) ((S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)) :*: (S1 ('MetaSel ('Just "runningMean") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 MutableTensor) :*: S1 ('MetaSel ('Just "runningVar") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 MutableTensor))))

batchNormForwardIO :: BatchNorm -> Bool -> Double -> Double -> Tensor -> IO Tensor Source #

data InstanceNormSpec Source #

Constructors

InstanceNormSpec 

Fields

Instances

Instances details
Show InstanceNormSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> InstanceNormSpec -> ShowS Source #

show :: InstanceNormSpec -> String Source #

showList :: [InstanceNormSpec] -> ShowS Source #

Eq InstanceNormSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: InstanceNormSpec -> InstanceNormSpec -> Bool Source #

(/=) :: InstanceNormSpec -> InstanceNormSpec -> Bool Source #

Randomizable InstanceNormSpec InstanceNorm Source # 
Instance details

Defined in Torch.NN

Methods

sample :: InstanceNormSpec -> IO InstanceNorm Source #

data InstanceNorm Source #

Constructors

InstanceNorm 

Fields

Instances

Instances details
Generic InstanceNorm Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep InstanceNorm :: Type -> Type Source #

Methods

from :: InstanceNorm -> Rep InstanceNorm x Source #

to :: Rep InstanceNorm x -> InstanceNorm Source #

Show InstanceNorm Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> InstanceNorm -> ShowS Source #

show :: InstanceNorm -> String Source #

showList :: [InstanceNorm] -> ShowS Source #

Randomizable InstanceNormSpec InstanceNorm Source # 
Instance details

Defined in Torch.NN

Methods

sample :: InstanceNormSpec -> IO InstanceNorm Source #

type Rep InstanceNorm Source # 
Instance details

Defined in Torch.NN

type Rep InstanceNorm = D1 ('MetaData "InstanceNorm" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "InstanceNorm" 'PrefixI 'True) ((S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)) :*: (S1 ('MetaSel ('Just "runningMean") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 MutableTensor) :*: S1 ('MetaSel ('Just "runningVar") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 MutableTensor))))

instanceNormForwardIO :: InstanceNorm -> Bool -> Double -> Double -> Tensor -> IO Tensor Source #

data UpSampleSpec Source #

Constructors

UpSampleSpec 

Fields

Instances

Instances details
Show UpSampleSpec Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> UpSampleSpec -> ShowS Source #

show :: UpSampleSpec -> String Source #

showList :: [UpSampleSpec] -> ShowS Source #

Eq UpSampleSpec Source # 
Instance details

Defined in Torch.NN

Methods

(==) :: UpSampleSpec -> UpSampleSpec -> Bool Source #

(/=) :: UpSampleSpec -> UpSampleSpec -> Bool Source #

Parameterized UpSampleSpec Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: UpSampleSpec -> [Parameter] Source #

_replaceParameters :: UpSampleSpec -> ParamStream UpSampleSpec Source #

Randomizable UpSampleSpec UpSample Source # 
Instance details

Defined in Torch.NN

Methods

sample :: UpSampleSpec -> IO UpSample Source #

data UpSample Source #

Constructors

UpSample 

Fields

Instances

Instances details
Generic UpSample Source # 
Instance details

Defined in Torch.NN

Associated Types

type Rep UpSample :: Type -> Type Source #

Methods

from :: UpSample -> Rep UpSample x Source #

to :: Rep UpSample x -> UpSample Source #

Show UpSample Source # 
Instance details

Defined in Torch.NN

Methods

showsPrec :: Int -> UpSample -> ShowS Source #

show :: UpSample -> String Source #

showList :: [UpSample] -> ShowS Source #

Parameterized UpSample Source # 
Instance details

Defined in Torch.NN

Methods

flattenParameters :: UpSample -> [Parameter] Source #

_replaceParameters :: UpSample -> ParamStream UpSample Source #

Randomizable UpSampleSpec UpSample Source # 
Instance details

Defined in Torch.NN

Methods

sample :: UpSampleSpec -> IO UpSample Source #

HasForward UpSample Tensor Tensor Source # 
Instance details

Defined in Torch.NN

Methods

forward :: UpSample -> Tensor -> Tensor Source #

forwardStoch :: UpSample -> Tensor -> IO Tensor Source #

type Rep UpSample Source # 
Instance details

Defined in Torch.NN

type Rep UpSample = D1 ('MetaData "UpSample" "Torch.NN" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "UpSample" 'PrefixI 'True) (S1 ('MetaSel ('Just "upsampleSpec") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 UpSampleSpec)))