Torch.Typed.NN.Sparse

Documentation

data EmbeddingType

Constructors

Constant

Learned

Instances

Generic EmbeddingType

Defined in Torch.Typed.NN.Sparse

Associated Types

type Rep EmbeddingType :: Type -> Type

Methods

from :: EmbeddingType -> Rep EmbeddingType x

to :: Rep EmbeddingType x -> EmbeddingType

Show EmbeddingType

Defined in Torch.Typed.NN.Sparse

Methods

showsPrec :: Int -> EmbeddingType -> ShowS

show :: EmbeddingType -> String

showList :: [EmbeddingType] -> ShowS

type Rep EmbeddingType

Defined in Torch.Typed.NN.Sparse

type Rep EmbeddingType = D1 ('MetaData "EmbeddingType" "Torch.Typed.NN.Sparse" "hasktorch-0.2.1.1-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

Constructors

ConstEmbeddingSpec :: forall paddingIdx numEmbeds embedSize dtype device. Tensor device dtype '[numEmbeds, embedSize] -> EmbeddingSpec paddingIdx numEmbeds embedSize 'Constant dtype device

LearnedEmbeddingWithRandomInitSpec :: forall paddingIdx numEmbeds embedSize dtype device. EmbeddingSpec paddingIdx numEmbeds embedSize 'Learned dtype device

LearnedEmbeddingWithCustomInitSpec :: forall paddingIdx numEmbeds embedSize dtype device. Tensor device dtype '[numEmbeds, embedSize] -> EmbeddingSpec paddingIdx numEmbeds embedSize 'Learned dtype device

Instances

Show (EmbeddingSpec paddingIdx numEmbeds embedSize embeddingType dtype device)

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)

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)

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

Defined in Torch.Typed.NN.Sparse

Methods

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

Randomizable (EmbeddingSpec paddingIdx numEmbeds embedSize 'Constant dtype device) (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device)

Defined in Torch.Typed.NN.Sparse

Methods

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

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

Constructors

ConstEmbedding

Fields

• :: forall paddingIdx numEmbeds embedSize dtype device. { constEmbedWeights :: Tensor device dtype '[numEmbeds, embedSize]
•    } -> Embedding paddingIdx numEmbeds embedSize 'Constant dtype device

LearnedEmbedding

Fields

• :: forall paddingIdx numEmbeds embedSize dtype device. { learnedEmbedWeights :: Parameter device dtype '[numEmbeds, embedSize]
•    } -> Embedding paddingIdx numEmbeds embedSize 'Learned dtype device

Instances

Generic (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device)

Defined in Torch.Typed.NN.Sparse

Associated Types

type Rep (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) :: Type -> Type

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)

Defined in Torch.Typed.NN.Sparse

Associated Types

type Rep (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) :: Type -> Type

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)

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)

Defined in Torch.Typed.NN.Sparse

Associated Types

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device) :: [Type]

Methods

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

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

Parameterized (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device)

Defined in Torch.Typed.NN.Sparse

Associated Types

type Parameters (Embedding paddingIdx numEmbeds embedSize 'Learned dtype device) :: [Type]

Methods

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

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

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

Defined in Torch.Typed.NN.Sparse

Methods

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

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

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

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)

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

Defined in Torch.Typed.NN.Sparse

Methods

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

Randomizable (EmbeddingSpec paddingIdx numEmbeds embedSize 'Constant dtype device) (Embedding paddingIdx numEmbeds embedSize 'Constant dtype device)

Defined in Torch.Typed.NN.Sparse

Methods

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

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

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)

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)

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)

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 shape numEmbeds embedSize embeddingType dtype device shape'. (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'