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

Torch.Typed.NN.Convolution

Synopsis

Documentation

data Conv1dSpec (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

Constructors

Conv1dSpec 

Instances

Instances details
Show (Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> ShowS Source #

show :: Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> String Source #

showList :: [Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device] -> ShowS Source #

Eq (Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

(==) :: Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> Bool Source #

(/=) :: Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> Conv1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> Bool 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 #

data Conv1d (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

Conv1d 

Fields

  • :: forall inputChannelSize outputChannelSize kernelSize dtype device. { weight :: Parameter device dtype '[outputChannelSize, inputChannelSize, kernelSize]
     
  •    , bias :: Parameter device dtype '[outputChannelSize]
     
  •    } -> Conv1d inputChannelSize outputChannelSize kernelSize dtype device
     

Instances

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Rep (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) :: Type -> Type Source #

Methods

from :: Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> Rep (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) x Source #

to :: Rep (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) x -> Conv1d inputChannelSize outputChannelSize kernelSize dtype device Source #

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

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> ShowS Source #

show :: Conv1d inputChannelSize outputChannelSize kernelSize dtype device -> String Source #

showList :: [Conv1d inputChannelSize outputChannelSize kernelSize dtype device] -> ShowS Source #

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) :: [Type] Source #

Methods

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

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

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

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

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

Defined in Torch.Typed.NN.Convolution

type Rep (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) = D1 ('MetaData "Conv1d" "Torch.Typed.NN.Convolution" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Conv1d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize, inputChannelSize, kernelSize])) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize]))))
type Parameters (Conv1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

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

conv1dForward :: forall stride padding. _ => Conv1d _ _ _ _ _ -> Tensor _ _ _ -> Tensor _ _ _ Source #

conv1d The constraints on this one are _very_ involved, so the partial signatures make the code significantly cleaner.

data Conv2dSpec (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

Constructors

Conv2dSpec 

Instances

Instances details
Show (Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> ShowS Source #

show :: Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> String Source #

showList :: [Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device] -> ShowS Source #

Eq (Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

(==) :: Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Bool Source #

(/=) :: Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Conv2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Bool 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 #

data Conv2d (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

Conv2d 

Fields

  • :: forall inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device. { weight :: Parameter device dtype '[outputChannelSize, inputChannelSize, kernelSize0, kernelSize1]
     
  •    , bias :: Parameter device dtype '[outputChannelSize]
     
  •    } -> Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device
     

Instances

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Rep (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) :: Type -> Type Source #

Methods

from :: Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Rep (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) x Source #

to :: Rep (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) x -> Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device Source #

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

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> ShowS Source #

show :: Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> String Source #

showList :: [Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device] -> ShowS Source #

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) :: [Type] Source #

Methods

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

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

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

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

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

Defined in Torch.Typed.NN.Convolution

type Rep (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) = D1 ('MetaData "Conv2d" "Torch.Typed.NN.Convolution" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Conv2d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize, inputChannelSize, kernelSize0, kernelSize1])) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize]))))
type Parameters (Conv2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

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

conv2dForward :: forall stride padding. _ => Conv2d _ _ _ _ _ _ -> Tensor _ _ _ -> Tensor _ _ _ Source #

conv2d The constraints on this one are _very_ involved, so the partial signatures make the code significantly cleaner.

data Conv3dSpec (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (kernelSize2 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

Constructors

Conv3dSpec 

Instances

Instances details
Show (Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> ShowS Source #

show :: Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> String Source #

showList :: [Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device] -> ShowS Source #

Eq (Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

(==) :: Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Bool Source #

(/=) :: Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Conv3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Bool 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 #

data Conv3d (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (kernelSize2 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

Conv3d 

Fields

  • :: forall inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device. { weight :: Parameter device dtype '[outputChannelSize, inputChannelSize, kernelSize0, kernelSize1, kernelSize2]
     
  •    , bias :: Parameter device dtype '[outputChannelSize]
     
  •    } -> Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device
     

Instances

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Rep (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) :: Type -> Type Source #

Methods

from :: Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Rep (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) x Source #

to :: Rep (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) x -> Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device Source #

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

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> ShowS Source #

show :: Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> String Source #

showList :: [Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device] -> ShowS Source #

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) :: [Type] Source #

Methods

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

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

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

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

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

Defined in Torch.Typed.NN.Convolution

type Rep (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) = D1 ('MetaData "Conv3d" "Torch.Typed.NN.Convolution" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "Conv3d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize, inputChannelSize, kernelSize0, kernelSize1, kernelSize2])) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize]))))
type Parameters (Conv3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

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

conv3dForward :: forall stride padding. _ => Conv3d _ _ _ _ _ _ _ -> Tensor _ _ _ -> Tensor _ _ _ Source #

conv3d The constraints on this one are _very_ involved, so the partial signatures make the code significantly cleaner.

data ConvTranspose1dSpec (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

Constructors

ConvTranspose1dSpec 

Instances

Instances details
Show (ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> ShowS Source #

show :: ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> String Source #

showList :: [ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device] -> ShowS Source #

Eq (ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

(==) :: ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> Bool Source #

(/=) :: ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> ConvTranspose1dSpec inputChannelSize outputChannelSize kernelSize dtype device -> Bool 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 #

data ConvTranspose1d (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

ConvTranspose1d 

Fields

  • :: forall inputChannelSize outputChannelSize kernelSize dtype device. { weight :: Parameter device dtype '[inputChannelSize, outputChannelSize, kernelSize]
     
  •    , bias :: Parameter device dtype '[outputChannelSize]
     
  •    } -> ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device
     

Instances

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Rep (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) :: Type -> Type Source #

Methods

from :: ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> Rep (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) x Source #

to :: Rep (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) x -> ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device Source #

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

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> ShowS Source #

show :: ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device -> String Source #

showList :: [ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device] -> ShowS Source #

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) :: [Type] Source #

Methods

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

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

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

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

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

Defined in Torch.Typed.NN.Convolution

type Rep (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) = D1 ('MetaData "ConvTranspose1d" "Torch.Typed.NN.Convolution" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "ConvTranspose1d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[inputChannelSize, outputChannelSize, kernelSize])) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize]))))
type Parameters (ConvTranspose1d inputChannelSize outputChannelSize kernelSize dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

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

convTranspose1dForward :: forall stride padding. _ => ConvTranspose1d _ _ _ _ _ -> Tensor _ _ _ -> Tensor _ _ _ Source #

convTranspose1d The constraints on this one are _very_ involved, so the partial signatures make the code significantly cleaner.

data ConvTranspose2dSpec (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

Constructors

ConvTranspose2dSpec 

Instances

Instances details
Show (ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> ShowS Source #

show :: ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> String Source #

showList :: [ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device] -> ShowS Source #

Eq (ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

(==) :: ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Bool Source #

(/=) :: ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> ConvTranspose2dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Bool 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 #

data ConvTranspose2d (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

ConvTranspose2d 

Fields

  • :: forall inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device. { weight :: Parameter device dtype '[inputChannelSize, outputChannelSize, kernelSize0, kernelSize1]
     
  •    , bias :: Parameter device dtype '[outputChannelSize]
     
  •    } -> ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device
     

Instances

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Rep (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) :: Type -> Type Source #

Methods

from :: ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> Rep (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) x Source #

to :: Rep (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) x -> ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device Source #

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

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> ShowS Source #

show :: ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device -> String Source #

showList :: [ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device] -> ShowS Source #

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) :: [Type] Source #

Methods

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

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

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

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

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

Defined in Torch.Typed.NN.Convolution

type Rep (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) = D1 ('MetaData "ConvTranspose2d" "Torch.Typed.NN.Convolution" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "ConvTranspose2d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[inputChannelSize, outputChannelSize, kernelSize0, kernelSize1])) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize]))))
type Parameters (ConvTranspose2d inputChannelSize outputChannelSize kernelSize0 kernelSize1 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

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

convTranspose2dForward :: forall stride padding. _ => ConvTranspose2d _ _ _ _ _ _ -> Tensor _ _ _ -> Tensor _ _ _ Source #

convTranspose2d The constraints on this one are _very_ involved, so the partial signatures make the code significantly cleaner.

data ConvTranspose3dSpec (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (kernelSize2 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

Constructors

ConvTranspose3dSpec 

Instances

Instances details
Show (ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> ShowS Source #

show :: ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> String Source #

showList :: [ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device] -> ShowS Source #

Eq (ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

Methods

(==) :: ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Bool Source #

(/=) :: ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> ConvTranspose3dSpec inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Bool 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 #

data ConvTranspose3d (inputChannelSize :: Nat) (outputChannelSize :: Nat) (kernelSize0 :: Nat) (kernelSize1 :: Nat) (kernelSize2 :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) where Source #

Constructors

ConvTranspose3d 

Fields

  • :: forall inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device. { weight :: Parameter device dtype '[inputChannelSize, outputChannelSize, kernelSize0, kernelSize1, kernelSize2]
     
  •    , bias :: Parameter device dtype '[outputChannelSize]
     
  •    } -> ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device
     

Instances

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Rep (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) :: Type -> Type Source #

Methods

from :: ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> Rep (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) x Source #

to :: Rep (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) x -> ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device Source #

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

Defined in Torch.Typed.NN.Convolution

Methods

showsPrec :: Int -> ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> ShowS Source #

show :: ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device -> String Source #

showList :: [ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device] -> ShowS Source #

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

Defined in Torch.Typed.NN.Convolution

Associated Types

type Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) :: [Type] Source #

Methods

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

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

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

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

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

Defined in Torch.Typed.NN.Convolution

type Rep (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) = D1 ('MetaData "ConvTranspose3d" "Torch.Typed.NN.Convolution" "hasktorch-0.2.0.0-F6yFRaDiRF49lpq95SVuR8" 'False) (C1 ('MetaCons "ConvTranspose3d" 'PrefixI 'True) (S1 ('MetaSel ('Just "weight") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[inputChannelSize, outputChannelSize, kernelSize0, kernelSize1, kernelSize2])) :*: S1 ('MetaSel ('Just "bias") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[outputChannelSize]))))
type Parameters (ConvTranspose3d inputChannelSize outputChannelSize kernelSize0 kernelSize1 kernelSize2 dtype device) Source # 
Instance details

Defined in Torch.Typed.NN.Convolution

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

convTranspose3dForward :: forall stride padding. _ => ConvTranspose3d _ _ _ _ _ _ _ -> Tensor _ _ _ -> Tensor _ _ _ Source #

convTranspose3d The constraints on this one are _very_ involved, so the partial signatures make the code significantly cleaner.