Torch.Typed.Tensor

Documentation

class KnownShape (shape :: [Nat]) where

Methods

shapeVal :: [Int]

Instances

KnownShape ('[] :: [Nat])

Defined in Torch.Typed.Tensor

Methods

shapeVal :: [Int]

(KnownNat h, KnownShape t) => KnownShape (h ': t)

Defined in Torch.Typed.Tensor

Methods

shapeVal :: [Int]

getFiniteI :: forall (n :: Nat). Finite n -> Int

class KnownDType (dtype :: DType) where

Methods

dtypeVal :: DType

Instances

KnownDType 'Bool

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'ComplexDouble

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'ComplexFloat

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'ComplexHalf

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Double

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Float

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Half

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Int16

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Int32

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Int64

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'Int8

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

KnownDType 'UInt8

Defined in Torch.Typed.Tensor

Methods

dtypeVal :: DType

type family ComputeDType (dtype' :: dtype) :: DType where ...

Equations

ComputeDType Bool = 'Bool

ComputeDType 'Bool = 'Bool

ComputeDType 'UInt8 = 'UInt8

ComputeDType 'Int8 = 'Int8

ComputeDType 'Int16 = 'Int16

ComputeDType 'Int32 = 'Int32

ComputeDType Int = 'Int64

ComputeDType 'Int64 = 'Int64

ComputeDType Half = 'Half

ComputeDType 'Half = 'Half

ComputeDType Float = 'Float

ComputeDType 'Float = 'Float

ComputeDType Double = 'Double

ComputeDType 'Double = 'Double

ComputeDType (Complex Half) = 'ComplexHalf

ComputeDType 'ComplexHalf = 'ComplexHalf

ComputeDType (Complex Float) = 'ComplexFloat

ComputeDType 'ComplexFloat = 'ComplexFloat

ComputeDType (Complex Double) = 'ComplexDouble

ComputeDType 'ComplexDouble = 'ComplexDouble

ComputeDType (dtype' :: t) = TypeError ('Text "Unsupported tensor type " ':<>: 'ShowType dtype') :: DType

class KnownDevice (device :: (DeviceType, Nat)) where

Methods

deviceVal :: Device

Instances

KnownNat n => KnownDevice '('CPU, n)

Defined in Torch.Typed.Tensor

Methods

deviceVal :: Device

KnownNat n => KnownDevice '('CUDA, n)

Defined in Torch.Typed.Tensor

Methods

deviceVal :: Device

KnownNat n => KnownDevice '('MPS, n)

Defined in Torch.Typed.Tensor

Methods

deviceVal :: Device

type Size = Type -> Type

type Shape = [Type -> Type]

type family ToNat (shape :: Size) :: Nat where ...

Equations

ToNat (S1 ('MetaSel _1 _2 _3 _4) f) = ToNat f

ToNat (D1 _1 f) = ToNat f

ToNat (C1 _1 f) = ToNat f

ToNat (l :*: r) = ToNat l + ToNat r

ToNat (l :+: r) = If (ToNat l <=? ToNat r) (ToNat r) (ToNat l)

ToNat (K1 R (Vector n _1) :: Type -> Type) = n

ToNat (K1 _1 _2 :: Type -> Type) = 1

ToNat (U1 :: Type -> Type) = 1

ToNat (Vector n) = n

ToNat a = ToNat (Rep (a ()))

type family ToNats (shape :: Shape) :: [Nat] where ...

Equations

ToNats ('[] :: [Type -> Type]) = '[] :: [Nat]

ToNats (x ': xs) = ToNat x ': ToNats xs

type family FromNat (shape :: Nat) :: Size where ...

Equations

FromNat n = Vector n

type family FromNats (shape :: [Nat]) :: Shape where ...

Equations

FromNats ('[] :: [Nat]) = '[] :: [Type -> Type]

FromNats (x ': xs) = FromNat x ': FromNats xs

class Unnamed t where

Associated Types

type UTShape t :: [Nat]

type UTDevice t :: (DeviceType, Nat)

type UTDType t :: DType

Methods

toUnnamed :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). IsUnnamed t device dtype shape => t -> Tensor device dtype shape

fromUnnamed :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). IsUnnamed t device dtype shape => Tensor device dtype shape -> t

toDynamic :: t -> Tensor

Instances

Unnamed (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

Associated Types

type UTShape (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTShape (NamedTensor device dtype shape) = ToNats shape

type UTDevice (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDevice (NamedTensor device dtype shape) = device

type UTDType (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDType (NamedTensor device dtype shape) = dtype

Methods

toUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (NamedTensor device dtype shape) device0 dtype0 shape0 => NamedTensor device dtype shape -> Tensor device0 dtype0 shape0

fromUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (NamedTensor device dtype shape) device0 dtype0 shape0 => Tensor device0 dtype0 shape0 -> NamedTensor device dtype shape

toDynamic :: NamedTensor device dtype shape -> Tensor

Unnamed (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

Associated Types

type UTShape (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTShape (Tensor device dtype shape) = shape

type UTDevice (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDevice (Tensor device dtype shape) = device

type UTDType (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDType (Tensor device dtype shape) = dtype

Methods

toUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (Tensor device dtype shape) device0 dtype0 shape0 => Tensor device dtype shape -> Tensor device0 dtype0 shape0

fromUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (Tensor device dtype shape) device0 dtype0 shape0 => Tensor device0 dtype0 shape0 -> Tensor device dtype shape

toDynamic :: Tensor device dtype shape -> Tensor

type family IsUnnamed t (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) where ...

Equations

IsUnnamed t device dtype shape = (Unnamed t, device ~ UTDevice t, dtype ~ UTDType t, shape ~ UTShape t)

data Tensor (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) where

Constructors

UnsafeMkTensor :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). Tensor -> Tensor device dtype shape

Instances

KnownDType dtype' => HasToDType (dtype' :: DType) (dtype :: DType) (Tensor device dtype shape) (Tensor device dtype' shape)

Defined in Torch.Typed.DType

Methods

toDType :: Tensor device dtype shape -> Tensor device dtype' shape

(KnownNat batchSize, KnownDevice device, Applicative m) => Dataset (m :: Type -> Type) (MNIST m device batchSize) Int ((Tensor device 'Float '[batchSize, 784], Tensor device 'Int64 '[batchSize]) :: Type)

Defined in Torch.Typed.Vision

Methods

getItem :: MNIST m device batchSize -> Int -> m (Tensor device 'Float '[batchSize, 784], Tensor device 'Int64 '[batchSize])

keys :: MNIST m device batchSize -> Set Int

KnownDevice device' => HasToDevice device' device (Tensor device dtype shape) (Tensor device' dtype shape)

Defined in Torch.Typed.Device

Methods

toDevice :: Tensor device dtype shape -> Tensor device' dtype shape

(HasGrad (HList parameters) (HList gradients), Castable (HList gradients) [ATenTensor]) => Apply' GradConcurrentlyF (HList parameters, Loss device dtype) (Concurrently (HList gradients))

Defined in Torch.Typed.NN.DataParallel

Methods

apply' :: GradConcurrentlyF -> (HList parameters, Loss device dtype) -> Concurrently (HList gradients)

Apply' MakeIndependent (Tensor device dtype shape) (IO (Parameter device dtype shape))

Defined in Torch.Typed.Parameter

Methods

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

Apply' ToParameter (Tensor dev dtype shape) (Parameter dev dtype shape)

Defined in Torch.Typed.Optim.CppOptim

Methods

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

Apply' ToDependent (Parameter device dtype shape) (Tensor device dtype shape)

Defined in Torch.Typed.Parameter

Methods

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

HasForward Dropout (Tensor device dtype shape) (Tensor device dtype shape)

Defined in Torch.Typed.NN.Dropout

Methods

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

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

(TensorLike [ComputeItemType (ComputeHaskellType dtype) shape], KnownDevice device, KnownShape shape) => IsList (Maybe (Tensor device dtype shape))

Defined in Torch.Typed.Tensor

Associated Types

type Item (Maybe (Tensor device dtype shape))

Defined in Torch.Typed.Tensor

type Item (Maybe (Tensor device dtype shape)) = ComputeItemType (ComputeHaskellType dtype) shape

Methods

fromList :: [Item (Maybe (Tensor device dtype shape))] -> Maybe (Tensor device dtype shape)

fromListN :: Int -> [Item (Maybe (Tensor device dtype shape))] -> Maybe (Tensor device dtype shape)

toList :: Maybe (Tensor device dtype shape) -> [Item (Maybe (Tensor device dtype shape))]

Castable [Tensor device dtype shape] (ForeignPtr TensorList)

Defined in Torch.Typed.Tensor

Methods

cast :: [Tensor device dtype shape] -> (ForeignPtr TensorList -> IO r) -> IO r

uncast :: ForeignPtr TensorList -> ([Tensor device dtype shape] -> IO r) -> IO r

(chunks ~ ListLength devices', tensorChunks ~ (Chunk chunks 0 shape dtype device :: [Type]), Castable (HList tensorChunks) [ATenTensor], devices ~ HReplicateR chunks device, HasToDevices devices' devices tensorChunks gs, KnownNat chunks) => HasScatter (devices' :: [(DeviceType, Nat)]) (device :: (DeviceType, Nat)) (Tensor device dtype shape) (gs :: [Type])

Defined in Torch.Typed.Device

Methods

scatter :: Tensor device dtype shape -> HList gs

KnownNat n => Castable (Vector n (Tensor device dtype shape)) (ForeignPtr TensorList)

Defined in Torch.Typed.Tensor

Methods

cast :: Vector n (Tensor device dtype shape) -> (ForeignPtr TensorList -> IO r) -> IO r

uncast :: ForeignPtr TensorList -> (Vector n (Tensor device dtype shape) -> IO r) -> IO r

(chunks ~ ListLength fs, devices ~ GetDevices fs, devices' ~ HReplicateR chunks device', HasToDevices devices' devices fs tensorChunks, '(shape, dtype, device') ~ Cat 0 tensorChunks, Castable (HList tensorChunks) [ATenTensor]) => HasGather (device' :: (DeviceType, Nat)) (devices :: [(DeviceType, Nat)]) (fs :: [Type]) (Tensor device' dtype shape)

Defined in Torch.Typed.Device

Methods

gather :: HList fs -> Tensor device' dtype shape

KnownDevice device => Num (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

Methods

(+) :: Tensor device dtype shape -> Tensor device dtype shape -> Tensor device dtype shape

(-) :: Tensor device dtype shape -> Tensor device dtype shape -> Tensor device dtype shape

(*) :: Tensor device dtype shape -> Tensor device dtype shape -> Tensor device dtype shape

negate :: Tensor device dtype shape -> Tensor device dtype shape

abs :: Tensor device dtype shape -> Tensor device dtype shape

signum :: Tensor device dtype shape -> Tensor device dtype shape

fromInteger :: Integer -> Tensor device dtype shape

KnownDevice device => Fractional (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

Methods

(/) :: Tensor device dtype shape -> Tensor device dtype shape -> Tensor device dtype shape

recip :: Tensor device dtype shape -> Tensor device dtype shape

fromRational :: Rational -> Tensor device dtype shape

Show (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

Methods

showsPrec :: Int -> Tensor device dtype shape -> ShowS

show :: Tensor device dtype shape -> String

showList :: [Tensor device dtype shape] -> ShowS

TensorOptions shape dtype device => Default (Tensor device dtype shape)

Defined in Torch.Typed.Factories

Methods

def :: Tensor device dtype shape

Parameterized (Tensor device dtype shape)

Defined in Torch.Typed.Parameter

Associated Types

type Parameters (Tensor device dtype shape)

Defined in Torch.Typed.Parameter

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

Methods

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

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

Unnamed (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

Associated Types

type UTShape (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTShape (Tensor device dtype shape) = shape

type UTDevice (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDevice (Tensor device dtype shape) = device

type UTDType (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDType (Tensor device dtype shape) = dtype

Methods

toUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (Tensor device dtype shape) device0 dtype0 shape0 => Tensor device dtype shape -> Tensor device0 dtype0 shape0

fromUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (Tensor device dtype shape) device0 dtype0 shape0 => Tensor device0 dtype0 shape0 -> Tensor device dtype shape

toDynamic :: Tensor device dtype shape -> Tensor

Castable (Tensor device dtype shape) ATenTensor

Defined in Torch.Typed.Tensor

Methods

cast :: Tensor device dtype shape -> (ATenTensor -> IO r) -> IO r

uncast :: ATenTensor -> (Tensor device dtype shape -> IO r) -> IO r

HasGrad (Parameter device dtype shape) (Tensor device dtype shape)

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

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

(IsSuffixOf normalizedShape shape, KnownShape normalizedShape) => HasForward (LayerNorm normalizedShape dtype device) (Tensor device dtype shape) (Tensor device dtype shape)

Defined in Torch.Typed.NN.Normalization

Methods

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

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

(1 <= numHeads, embedDim ~ (headDim * numHeads), All KnownNat '[embedDim, numHeads, seqLen, batchSize, headDim], IsSuffixOf '[embedDim] '[batchSize, seqLen, embedDim], KnownDType dtype, StandardFloatingPointDTypeValidation device dtype, MatMulDTypeIsValid device dtype, BasicArithmeticDTypeIsValid device dtype, dtype ~ SumDType dtype, SumDTypeIsValid device dtype, KnownDevice device) => Apply' (FoldLayers batchSize seqLen dtype device) (TransformerLayer embedDim embedDim embedDim numHeads ffnDim dtype device, IO (Tensor device dtype '[batchSize, seqLen, embedDim])) (IO (Tensor device dtype '[batchSize, seqLen, embedDim]))

Defined in Torch.Typed.NN.Transformer

Methods

apply' :: FoldLayers batchSize seqLen dtype device -> (TransformerLayer embedDim embedDim embedDim numHeads ffnDim dtype device, IO (Tensor device dtype '[batchSize, seqLen, embedDim])) -> IO (Tensor device dtype '[batchSize, seqLen, embedDim])

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

Defined in Torch.Typed.NN.Linear

Methods

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

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

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

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]

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

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

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]

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

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

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]

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

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

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]

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

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

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

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]

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

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

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]

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

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

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]

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

type Item (Maybe (Tensor device dtype shape))

Defined in Torch.Typed.Tensor

type Item (Maybe (Tensor device dtype shape)) = ComputeItemType (ComputeHaskellType dtype) shape

type Parameters (Tensor device dtype shape)

Defined in Torch.Typed.Parameter

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

type UTDType (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDType (Tensor device dtype shape) = dtype

type UTDevice (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDevice (Tensor device dtype shape) = device

type UTShape (Tensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTShape (Tensor device dtype shape) = shape

type CPUTensor = Tensor '('CPU, 0)

type CUDATensor (deviceIndex :: Nat) = Tensor '('CUDA, deviceIndex)

type MPSTensor (deviceIndex :: k) = Tensor '('MPS, 0)

data UnknownShapeTensor (device :: (DeviceType, Nat)) (dtype :: DType)

Constructors

UnknownShapeTensor (Tensor device dtype shape)

type family ComputeHaskellType (dtype :: DType) where ...

Equations

ComputeHaskellType 'Bool = Bool

ComputeHaskellType 'Int64 = Int

ComputeHaskellType 'Float = Float

ComputeHaskellType 'Double = Double

ComputeHaskellType dtype = TypeError ('Text "Unsupported tensor type " ':<>: 'ShowType dtype) :: Type

type family ComputeItemType ty (shape :: [Nat]) where ...

Equations

ComputeItemType _1 ('[] :: [Nat]) = TypeError ('Text "Scalars are not supported") :: Type

ComputeItemType ty '[_1] = ty

ComputeItemType ty (_1 ': (h ': t)) = [ComputeItemType ty (h ': t)]

class TensorOptions (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)) where

Methods

optionsRuntimeShape :: [Int]

optionsRuntimeDType :: DType

optionsRuntimeDevice :: Device

Instances

(KnownDType dtype, KnownDevice device) => TensorOptions ('[] :: [Nat]) dtype device

Defined in Torch.Typed.Tensor

Methods

optionsRuntimeShape :: [Int]

optionsRuntimeDType :: DType

optionsRuntimeDevice :: Device

(KnownNat h, TensorOptions t dtype device) => TensorOptions (h ': t) dtype device

Defined in Torch.Typed.Tensor

Methods

optionsRuntimeShape :: [Int]

optionsRuntimeDType :: DType

optionsRuntimeDevice :: Device

type family All (pred :: a -> Constraint) (l :: [a]) where ...

Equations

All (_1 :: a -> Constraint) ('[] :: [a]) = ()

All (pred :: a -> Constraint) (h ': t :: [a]) = (pred h, All pred t)

data SomeShape where

Constructors

SomeShape :: forall (shape :: [Nat]). KnownShape shape => Proxy shape -> SomeShape

someShape :: [Int] -> SomeShape

data SomeDType where

Constructors

SomeDType :: forall (dtype :: DType). KnownDType dtype => Proxy dtype -> SomeDType

someDType :: DType -> SomeDType

data SomeDevice where

Constructors

SomeDevice :: forall (device :: (DeviceType, Nat)). KnownDevice device => Proxy device -> SomeDevice

someDevice :: Device -> SomeDevice

withTensor :: Tensor -> (forall (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDevice device, KnownDType dtype, KnownShape shape) => Tensor device dtype shape -> r) -> r

withTensorShape :: forall (device :: (DeviceType, Nat)) (dtype :: DType) r. (KnownDevice device, KnownDType dtype) => Tensor -> (forall (shape :: [Nat]). KnownShape shape => Tensor device dtype shape -> r) -> r

type family ComputeBroadcast (reversedShape :: [Nat]) (reversedShape' :: [Nat]) :: Maybe [Nat] where ...

Equations

ComputeBroadcast ('[] :: [Nat]) reversedShape = 'Just reversedShape

ComputeBroadcast reversedShape ('[] :: [Nat]) = 'Just reversedShape

ComputeBroadcast (h ': t) (h ': t2) = AppendToMaybe h (ComputeBroadcast t t2)

ComputeBroadcast (h ': t) (1 ': t2) = AppendToMaybe h (ComputeBroadcast t t2)

ComputeBroadcast (1 ': t) (h ': t2) = AppendToMaybe h (ComputeBroadcast t t2)

ComputeBroadcast _1 _2 = 'Nothing :: Maybe [Nat]

type family CheckBroadcast (shape :: [Nat]) (shape' :: [Nat]) (result :: Maybe [Nat]) :: [Nat] where ...

Equations

CheckBroadcast shape shape' ('Nothing :: Maybe [Nat]) = TypeError (((('Text "The shapes " ':<>: 'ShowType shape) ':<>: 'Text " and ") ':<>: 'ShowType shape') ':<>: 'Text " cannot be broadcast") :: [Nat]

CheckBroadcast _1 _2 ('Just result) = Reverse result

type Broadcast (shape :: [Nat]) (shape' :: [Nat]) = CheckBroadcast shape shape' (ComputeBroadcast (Reverse shape) (Reverse shape'))

type family BasicArithmeticDTypeIsValid (device :: (DeviceType, Nat)) (dtype :: DType) where ...

Equations

BasicArithmeticDTypeIsValid '('CPU, 0) dtype = (DTypeIsNotBool '('CPU, 0) dtype, DTypeIsNotHalf '('CPU, 0) dtype)

BasicArithmeticDTypeIsValid '('CUDA, _1) dtype = ()

BasicArithmeticDTypeIsValid '('MPS, 0) dtype = ()

BasicArithmeticDTypeIsValid '(deviceType, _1) dtype = UnsupportedDTypeForDevice deviceType dtype

add :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (dtype'' :: DType) (device :: (DeviceType, Nat)). (dtype'' ~ DTypePromotion dtype dtype', shape'' ~ Broadcast shape shape', BasicArithmeticDTypeIsValid device dtype, BasicArithmeticDTypeIsValid device dtype', BasicArithmeticDTypeIsValid device dtype'') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device dtype'' shape''

sub :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (dtype'' :: DType) (device :: (DeviceType, Nat)). (dtype'' ~ DTypePromotion dtype dtype', shape'' ~ Broadcast shape shape', BasicArithmeticDTypeIsValid device dtype, BasicArithmeticDTypeIsValid device dtype', BasicArithmeticDTypeIsValid device dtype'') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device dtype'' shape''

mul :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (dtype'' :: DType) (device :: (DeviceType, Nat)). (dtype'' ~ DTypePromotion dtype dtype', shape'' ~ Broadcast shape shape', BasicArithmeticDTypeIsValid device dtype, BasicArithmeticDTypeIsValid device dtype', BasicArithmeticDTypeIsValid device dtype'') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device dtype'' shape''

div :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (dtype'' :: DType) (device :: (DeviceType, Nat)). (dtype'' ~ DTypePromotion dtype dtype', shape'' ~ Broadcast shape shape', BasicArithmeticDTypeIsValid device dtype, BasicArithmeticDTypeIsValid device dtype', BasicArithmeticDTypeIsValid device dtype'') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device dtype'' shape''

type family ComparisonDTypeIsValid (device :: (DeviceType, Nat)) (dtype :: DType) where ...

Equations

ComparisonDTypeIsValid '('CPU, 0) dtype = (DTypeIsNotBool '('CPU, 0) dtype, DTypeIsNotHalf '('CPU, 0) dtype)

ComparisonDTypeIsValid '('CUDA, _1) dtype = ()

ComparisonDTypeIsValid '('MPS, 0) dtype = ()

ComparisonDTypeIsValid '(deviceType, _1) dtype = UnsupportedDTypeForDevice deviceType dtype

gt :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

lt :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

ge :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

le :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

eq :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

ne :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

(>.) :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

(<.) :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

(>=.) :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

(<=.) :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

(==.) :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

(/=.) :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (dtype' :: DType) (device :: (DeviceType, Nat)). (shape'' ~ Broadcast shape shape', ComparisonDTypeIsValid device dtype, ComparisonDTypeIsValid device dtype') => Tensor device dtype shape -> Tensor device dtype' shape' -> Tensor device 'Bool shape''

type family ComputeMatMul (reversedShape :: [Nat]) (reversedShape' :: [Nat]) :: Maybe [Nat] where ...

Equations

ComputeMatMul '[k] '[k] = 'Just ('[] :: [Nat])

ComputeMatMul '[k] (m ': (k ': reversedBroadcastShape')) = AppendToMaybe m (ComputeBroadcast ('[] :: [Nat]) reversedBroadcastShape')

ComputeMatMul (k ': (n ': reversedBroadcastShape)) '[k] = AppendToMaybe n (ComputeBroadcast ('[] :: [Nat]) reversedBroadcastShape)

ComputeMatMul (k ': (n ': reversedBroadcastShape)) (m ': (k ': reversedBroadcastShape')) = AppendToMaybe m (AppendToMaybe n (ComputeBroadcast reversedBroadcastShape reversedBroadcastShape'))

type family CheckMatMul (shape :: [Nat]) (shape' :: [Nat]) (result :: Maybe [Nat]) :: [Nat] where ...

Equations

CheckMatMul shape shape' ('Nothing :: Maybe [Nat]) = TypeError (((('Text "The shapes " ':<>: 'ShowType shape) ':<>: 'Text " and ") ':<>: 'ShowType shape') ':<>: 'Text " are not compatible with matrix multiplication") :: [Nat]

CheckMatMul _1 _2 ('Just result) = Reverse result

type MatMul (shape :: [Nat]) (shape' :: [Nat]) = CheckMatMul shape shape' (ComputeMatMul (Reverse shape) (Reverse shape'))

type family MatMulDTypeIsValid (device :: (DeviceType, Nat)) (dtype :: DType) where ...

Equations

MatMulDTypeIsValid '('CPU, 0) dtype = (DTypeIsNotBool '('CPU, 0) dtype, DTypeIsNotHalf '('CPU, 0) dtype)

MatMulDTypeIsValid '('CUDA, deviceIndex) dtype = DTypeIsFloatingPoint '('CUDA, deviceIndex) dtype

MatMulDTypeIsValid '('MPS, 0) dtype = DTypeIsFloatingPoint '('MPS, 0) dtype

MatMulDTypeIsValid '(deviceType, _1) dtype = UnsupportedDTypeForDevice deviceType dtype

matmul :: forall (shape'' :: [Nat]) (shape :: [Nat]) (shape' :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). (shape'' ~ MatMul shape shape', MatMulDTypeIsValid device dtype) => Tensor device dtype shape -> Tensor device dtype shape' -> Tensor device dtype shape''

matrix multiplication See https://pytorch.org/docs/stable/torch.html#torch.matmul.

select :: forall (dim :: Nat) (idx :: Nat) (shape' :: [Nat]) (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownNat dim, KnownNat idx, InRange shape dim idx, shape' ~ Remove shape dim) => Tensor device dtype shape -> Tensor device dtype shape'

selectIdx :: forall (dim :: Nat) (n :: Nat) (shape' :: [Nat]) (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownNat dim, n ~ Index shape dim, shape' ~ Remove shape dim) => Tensor device dtype shape -> Finite n -> Tensor device dtype shape'

type family Numel (shape :: [Nat]) :: Nat where ...

Equations

Numel ('[] :: [Nat]) = 1

Numel (h ': t) = h * Numel t

reshape :: forall (shape' :: [Nat]) (shape :: [Nat]) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownShape shape', Numel shape ~ Numel shape') => Tensor device dtype shape -> Tensor device dtype shape'

reshape >>> t :: CPUTensor 'D.Int64 '[2,3,4] = fromJust [[[111,112,113,114],[121,122,123,124],[131,132,133,134]],[[211,212,213,214],[221,222,223,224],[231,232,233,234]]] >>> t' = reshape '[24] t >>> toList . Just $ t' [111,112,113,114,121,122,123,124,131,132,133,134,211,212,213,214,221,222,223,224,231,232,233,234] >>> toList . Just $ reshape '[2,3,4] t' [[[111,112,113,114],[121,122,123,124],[131,132,133,134]],[[211,212,213,214],[221,222,223,224],[231,232,233,234]]]

newtype Wrap a

To avoid overlapped instance for (Unnamed t => Castable t D.ATenTensor)

Constructors

Wrap

Fields

• unWrap :: a

Instances

Unnamed t => Castable (Wrap t) ATenTensor

Defined in Torch.Typed.Tensor

Methods

cast :: Wrap t -> (ATenTensor -> IO r) -> IO r

uncast :: ATenTensor -> (Wrap t -> IO r) -> IO r

data TensorListFold

Constructors

TensorListFold

Instances

Castable x ATenTensor => Apply' TensorListFold (x, IO [ATenTensor]) (IO [ATenTensor])

Defined in Torch.Typed.Tensor

Methods

apply' :: TensorListFold -> (x, IO [ATenTensor]) -> IO [ATenTensor]

data TensorListUnfold

Constructors

TensorListUnfold

Instances

Castable x ATenTensor => Apply TensorListUnfold [ATenTensor] (IO (HJust (x, [ATenTensor])))

Defined in Torch.Typed.Tensor

Methods

apply :: TensorListUnfold -> [ATenTensor] -> IO (HJust (x, [ATenTensor]))

Apply TensorListUnfold [ATenTensor] (IO HNothing)

Defined in Torch.Typed.Tensor

Methods

apply :: TensorListUnfold -> [ATenTensor] -> IO HNothing

toSparse :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). Tensor device dtype shape -> Tensor device dtype shape

toDense :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). Tensor device dtype shape -> Tensor device dtype shape

toCPU :: forall (device :: (DeviceType, Nat)) (shape :: [Nat]) (dtype :: DType). Tensor device dtype shape -> CPUTensor dtype shape

move tensor to CPU TODO: can this fail?

toCUDA :: forall {k} (device' :: k) (device :: (DeviceType, Nat)) (shape :: [Nat]) (dtype :: DType). Tensor device dtype shape -> CUDATensor 0 dtype shape

move tensor to the first CUDA device TODO: what if this fails?

toMPS :: forall {k} (device' :: k) (device :: (DeviceType, Nat)) (shape :: [Nat]) (dtype :: DType). Tensor device dtype shape -> MPSTensor 0 dtype shape

move tensor to the first MPS device TODO: what if this fails?

toDevice :: forall (device' :: (DeviceType, Nat)) (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) t t'. (KnownDevice device', IsUnnamed t device dtype shape, Unnamed t', t' ~ ReplaceDevice'' t device') => t -> t'

move tensor to device TODO: what if this fails?

toDType :: forall (dtype' :: DType) (dtype :: DType) (device :: (DeviceType, Nat)) (shape :: [Nat]) t t'. (KnownDType dtype', IsUnnamed t device dtype shape, Unnamed t', t' ~ ReplaceDType'' t dtype') => t -> t'

change tensor data type

dim :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) t. (TensorOptions shape dtype device, IsUnnamed t device dtype shape) => t -> Int

returns tensor dimension uses compile-time information only

shape :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) t. (TensorOptions shape dtype device, IsUnnamed t device dtype shape) => t -> [Int]

returns tensor shape as list uses compile-time information only

dtype :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) t. (TensorOptions shape dtype device, IsUnnamed t device dtype shape) => t -> DType

returns tensor data type uses compile-time information only

device :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]) t. (TensorOptions shape dtype device, IsUnnamed t device dtype shape) => t -> Device

returns tensor device uses compile-time information only

toInt :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). Tensor device dtype shape -> Int

toFloat :: forall (device :: (DeviceType, Nat)). Tensor device 'Float ('[] :: [Nat]) -> Float

toDouble :: forall (device :: (DeviceType, Nat)). Tensor device 'Double ('[] :: [Nat]) -> Double

toBool :: forall (device :: (DeviceType, Nat)). Tensor device 'Bool ('[] :: [Nat]) -> Bool

type family ToDType a :: DType where ...

Equations

ToDType Bool = 'Bool

ToDType Int = 'Int64

ToDType Float = 'Float

ToDType Double = 'Double

ToDType (f a) = ToDType a

type family ToShape a :: Shape where ...

Equations

ToShape Bool = '[] :: [Type -> Type]

ToShape Int = '[] :: [Type -> Type]

ToShape Float = '[] :: [Type -> Type]

ToShape Double = '[] :: [Type -> Type]

ToShape (f a) = f ': ToShape a

type family FindDim (a :: Size) (shape :: Shape) :: Nat where ...

Equations

FindDim a (a ': _1) = 0

FindDim a (b ': ax) = 1 + FindDim a ax

FindDim a _1 = TypeError (('Text "Not find a type:" ':<>: 'ShowType a) ':<>: 'Text " in the shape.") :: Nat

data NamedTensor (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: Shape) where

Constructors

FromTensor :: forall (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: Shape) (shape1 :: [Nat]). shape1 ~ ToNats shape => Tensor device dtype shape1 -> NamedTensor device dtype shape

Instances

KnownDevice device => Num (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

Methods

(+) :: NamedTensor device dtype shape -> NamedTensor device dtype shape -> NamedTensor device dtype shape

(-) :: NamedTensor device dtype shape -> NamedTensor device dtype shape -> NamedTensor device dtype shape

(*) :: NamedTensor device dtype shape -> NamedTensor device dtype shape -> NamedTensor device dtype shape

negate :: NamedTensor device dtype shape -> NamedTensor device dtype shape

abs :: NamedTensor device dtype shape -> NamedTensor device dtype shape

signum :: NamedTensor device dtype shape -> NamedTensor device dtype shape

fromInteger :: Integer -> NamedTensor device dtype shape

KnownDevice device => Fractional (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

Methods

(/) :: NamedTensor device dtype shape -> NamedTensor device dtype shape -> NamedTensor device dtype shape

recip :: NamedTensor device dtype shape -> NamedTensor device dtype shape

fromRational :: Rational -> NamedTensor device dtype shape

Show (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

Methods

showsPrec :: Int -> NamedTensor device dtype shape -> ShowS

show :: NamedTensor device dtype shape -> String

showList :: [NamedTensor device dtype shape] -> ShowS

(TensorOptions shape' dtype device, shape' ~ ToNats shape) => Default (NamedTensor device dtype shape)

Defined in Torch.Typed.Factories

Methods

def :: NamedTensor device dtype shape

Unnamed (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

Associated Types

type UTShape (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTShape (NamedTensor device dtype shape) = ToNats shape

type UTDevice (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDevice (NamedTensor device dtype shape) = device

type UTDType (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDType (NamedTensor device dtype shape) = dtype

Methods

toUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (NamedTensor device dtype shape) device0 dtype0 shape0 => NamedTensor device dtype shape -> Tensor device0 dtype0 shape0

fromUnnamed :: forall (device0 :: (DeviceType, Nat)) (dtype0 :: DType) (shape0 :: [Nat]). IsUnnamed (NamedTensor device dtype shape) device0 dtype0 shape0 => Tensor device0 dtype0 shape0 -> NamedTensor device dtype shape

toDynamic :: NamedTensor device dtype shape -> Tensor

Castable (NamedTensor device dtype shape) ATenTensor

Defined in Torch.Typed.Tensor

Methods

cast :: NamedTensor device dtype shape -> (ATenTensor -> IO r) -> IO r

uncast :: ATenTensor -> (NamedTensor device dtype shape -> IO r) -> IO r

type UTDType (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDType (NamedTensor device dtype shape) = dtype

type UTDevice (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTDevice (NamedTensor device dtype shape) = device

type UTShape (NamedTensor device dtype shape)

Defined in Torch.Typed.Tensor

type UTShape (NamedTensor device dtype shape) = ToNats shape

type family ReplaceDevice'' (tensor :: t) (device :: (DeviceType, Nat)) :: t where ...

Equations

ReplaceDevice'' (Tensor device0 dtype shape :: Type) device1 = Tensor device1 dtype shape

ReplaceDevice'' (NamedTensor device0 dtype shape :: Type) device1 = NamedTensor device1 dtype shape

type family ReplaceDType'' (tensor :: t) (dtype :: DType) :: t where ...

Equations

ReplaceDType'' (Tensor device dtype0 shape :: Type) dtype1 = Tensor device dtype1 shape

ReplaceDType'' (NamedTensor device dtype0 shape :: Type) dtype1 = NamedTensor device dtype1 shape

Orphan instances

(HFoldrM IO TensorListFold [ATenTensor] l [ATenTensor], Apply TensorListUnfold [ATenTensor] res, HUnfoldM IO TensorListUnfold res l, res ~ HUnfoldMRes IO [ATenTensor] l) => Castable (HList l) [ATenTensor]

Methods

cast :: HList l -> ([ATenTensor] -> IO r) -> IO r

uncast :: [ATenTensor] -> (HList l -> IO r) -> IO r

Castable (HList l) [ATenTensor] => Castable (HList l) (ForeignPtr TensorList)

Methods

cast :: HList l -> (ForeignPtr TensorList -> IO r) -> IO r

uncast :: ForeignPtr TensorList -> (HList l -> IO r) -> IO r