Safe Haskell	None
Language	Haskell2010

Torch.Typed.NN.Sparse

Documentation

data EmbeddingType Source #

Constructors

Constant
Learned

Instances

Instances details

Generic EmbeddingType Source #

Instance details

Defined in Torch.Typed.NN.Sparse

Associated Types

type Rep EmbeddingType
Instance details Defined in Torch.Typed.NN.Sparse type Rep EmbeddingType = D1 ('MetaData "EmbeddingType" "Torch.Typed.NN.Sparse" "hasktorch-0.2.1.3-inplace" 'False) (C1 ('MetaCons "Constant" 'PrefixI 'False) (U1 :: Type -> Type) :+: C1 ('MetaCons "Learned" 'PrefixI 'False) (U1 :: Type -> Type))

Methods

from :: EmbeddingType -> Rep EmbeddingType x #

to :: Rep EmbeddingType x -> EmbeddingType #

Show EmbeddingType Source #

Instance details

Defined in Torch.Typed.NN.Sparse

Methods

showsPrec :: Int -> EmbeddingType -> ShowS #

show :: EmbeddingType -> String #

showList :: [EmbeddingType] -> ShowS #

type Rep EmbeddingType Source #

Instance details

Defined in Torch.Typed.NN.Sparse

type Rep EmbeddingType = D1 ('MetaData "EmbeddingType" "Torch.Typed.NN.Sparse" "hasktorch-0.2.1.3-inplace" 'False) (C1 ('MetaCons "Constant" 'PrefixI 'False) (U1 :: Type -> Type) :+: C1 ('MetaCons "Learned" 'PrefixI 'False) (U1 :: Type -> Type))

data EmbeddingSpec (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (embeddingType :: EmbeddingType) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

ConstEmbeddingSpec :: forall (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). Tensor device dtype '[numEmbeds, embedSize] -> EmbeddingSpec paddingIdx numEmbeds embedSize 'Constant dtype device
LearnedEmbeddingWithRandomInitSpec :: forall (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). EmbeddingSpec paddingIdx numEmbeds embedSize 'Learned dtype device
LearnedEmbeddingWithCustomInitSpec :: forall (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). Tensor device dtype '[numEmbeds, embedSize] -> EmbeddingSpec paddingIdx numEmbeds embedSize 'Learned dtype device

Instances

Instances details

Show (EmbeddingSpec paddingIdx numEmbeds embedSize embeddingType dtype device) Source #
Instance details Defined in Torch.Typed.NN.Sparse Methods showsPrec :: Int -> EmbeddingSpec paddingIdx numEmbeds embedSize embeddingType dtype device -> ShowS # show :: EmbeddingSpec paddingIdx numEmbeds embedSize embeddingType dtype device -> String # showList :: [EmbeddingSpec paddingIdx numEmbeds embedSize embeddingType dtype device] -> ShowS #
(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 #

data Embedding (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (embeddingType :: EmbeddingType) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

ConstEmbedding
Fields :: forall (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). { constEmbedWeights :: Tensor device dtype '[numEmbeds, embedSize] } -> Embedding paddingIdx numEmbeds embedSize 'Constant dtype device
LearnedEmbedding
Fields :: forall (paddingIdx :: Maybe Nat) (numEmbeds :: Nat) (embedSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). { learnedEmbedWeights :: Parameter device dtype '[numEmbeds, embedSize] } -> Embedding paddingIdx numEmbeds embedSize 'Learned dtype device

Instances

Instances details

Generic (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Sparse

Associated Types

type Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device)
Instance details Defined in Torch.Typed.NN.Sparse type Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) = Rec0 (Tensor device dtype '[numEmbeds, embedSize])

Methods

from :: Embedding paddingIdx numEmbeds embedSize 'Constant dtype device -> Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) x #

to :: Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) x -> Embedding paddingIdx numEmbeds embedSize 'Constant dtype device #

Generic (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Sparse

Associated Types

type Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device)
Instance details Defined in Torch.Typed.NN.Sparse type Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) = Rec0 (Parameter device dtype '[numEmbeds, embedSize])

Methods

from :: Embedding paddingIdx numEmbeds embedSize 'Learned dtype device -> Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) x #

to :: Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) x -> Embedding paddingIdx numEmbeds embedSize 'Learned dtype device #

Show (Embedding paddingIdx numEmbeds embedSize embeddingType dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Sparse

Methods

showsPrec :: Int -> Embedding paddingIdx numEmbeds embedSize embeddingType dtype device -> ShowS #

show :: Embedding paddingIdx numEmbeds embedSize embeddingType dtype device -> String #

showList :: [Embedding paddingIdx numEmbeds embedSize embeddingType dtype device] -> ShowS #

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 #

(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 #

(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 #

type Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Sparse

type Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) = Rec0 (Tensor device dtype '[numEmbeds, embedSize])

type Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Sparse

type Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) = Rec0 (Parameter device dtype '[numEmbeds, embedSize])

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))

embed :: forall (paddingIdx :: Maybe Nat) (shape :: [Nat]) (numEmbeds :: Nat) (embedSize :: Nat) (embeddingType :: EmbeddingType) (dtype :: DType) (device :: (DeviceType, Nat)) (shape' :: [Nat]). (KnownMaybeNat paddingIdx, PaddingIdxCheck paddingIdx numEmbeds, shape' ~ Reverse (embedSize ': Reverse shape)) => Embedding paddingIdx numEmbeds embedSize embeddingType dtype device -> Tensor device 'Int64 shape -> Tensor device dtype shape' Source #