Torch.Typed.Lens

Documentation

class HasName (name :: Type -> Type) shape where

Minimal complete definition

Nothing

Methods

name :: Traversal' (NamedTensor device dtype shape) (NamedTensor device dtype (DropName name shape))

default name :: KnownNat (NamedIdx name shape) => Traversal' (NamedTensor device dtype shape) (NamedTensor device dtype (DropName name shape))

Instances

KnownNat (NamedIdx name shape) => HasName name shape

Defined in Torch.Typed.Lens

Methods

name :: forall (device :: (DeviceType, Nat)) (dtype :: DType). Traversal' (NamedTensor device dtype shape) (NamedTensor device dtype (DropName name shape))

class HasField (field :: Symbol) shape where

Minimal complete definition

Nothing

Methods

field :: Lens' (NamedTensor device dtype shape) (NamedTensor device dtype (DropField field shape))

default field :: FieldIdx field shape => Lens' (NamedTensor device dtype shape) (NamedTensor device dtype (DropField field shape))

Instances

FieldIdx field shape => HasField field shape

Defined in Torch.Typed.Lens

Methods

field :: forall (device :: (DeviceType, Nat)) (dtype :: DType). Lens' (NamedTensor device dtype shape) (NamedTensor device dtype (DropField field shape))

type family GHasField (field :: Symbol) f :: Bool where ...

Equations

GHasField field (S1 ('MetaSel ('Just field) _ _ _) _) = 'True

GHasField field (S1 ('MetaSel _ _ _ _) _) = 'False

GHasField field (D1 _ f) = GHasField field f

GHasField field (C1 _ f) = GHasField field f

GHasField field (l :*: r) = GHasField field l || GHasField field r

GHasField field (l :+: r) = GHasField field l || GHasField field r

GHasField field (K1 _ _) = 'False

GHasField field U1 = 'False

GHasField field (Vector n) = 'False

GHasField field a = GHasField field (Rep (a ()))

type family DropField (field :: Symbol) (a :: [Type -> Type]) :: [Type -> Type] where ...

Equations

DropField field '[] = '[]

DropField field (x ': xs) = If (GHasField field x) xs (x ': DropField field xs)

type family DropName (name :: Type -> Type) (a :: [Type -> Type]) :: [Type -> Type] where ...

Equations

DropName name '[] = '[]

DropName name (name ': xs) = xs

DropName name (x ': xs) = x ': DropName name xs

type family NamedIdx (name :: Type -> Type) (shape :: [Type -> Type]) :: Nat where ...

Equations

NamedIdx name '[] = TypeError (Text "There is not the name in the shape.")

NamedIdx name (name ': xs) = 0

NamedIdx name (x ': xs) = NamedIdx name xs + 1

class FieldIdx (field :: Symbol) (a :: [Type -> Type]) where

Methods

fieldIdx :: Proxy a -> [Maybe Int]

Return field-id

Instances

FieldIdx field ('[] :: [Type -> Type])

Defined in Torch.Typed.Lens

Methods

fieldIdx :: Proxy '[] -> [Maybe Int]

(FieldId field (x ()), FieldIdx field xs) => FieldIdx field (x ': xs)

Defined in Torch.Typed.Lens

Methods

fieldIdx :: Proxy (x ': xs) -> [Maybe Int]

class FieldId (field :: Symbol) a where

Minimal complete definition

Nothing

Methods

fieldId :: Proxy a -> Maybe Int

Return field-id

default fieldId :: (Generic a, GFieldId field (Rep a)) => Proxy a -> Maybe Int

Instances

(Generic s, GFieldId field (Rep s)) => FieldId field s

Defined in Torch.Typed.Lens

Methods

fieldId :: Proxy s -> Maybe Int

FieldId field (Vector n v)

Defined in Torch.Typed.Lens

Methods

fieldId :: Proxy (Vector n v) -> Maybe Int

class GFieldId (field :: Symbol) (a :: Type -> Type) where

Minimal complete definition

gfieldId'

Methods

gfieldId :: Proxy a -> Maybe Int

gfieldId' :: Proxy a -> (Maybe Int, Int)

Instances

GFieldId field (U1 :: Type -> Type)

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy U1 -> Maybe Int

gfieldId' :: Proxy U1 -> (Maybe Int, Int)

(GFieldId field f, GFieldId field g) => GFieldId field (f :*: g)

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy (f :*: g) -> Maybe Int

gfieldId' :: Proxy (f :*: g) -> (Maybe Int, Int)

(GFieldId field f, GFieldId field g) => GFieldId field (f :+: g)

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy (f :+: g) -> Maybe Int

gfieldId' :: Proxy (f :+: g) -> (Maybe Int, Int)

GFieldId field (K1 c f :: Type -> Type)

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy (K1 c f) -> Maybe Int

gfieldId' :: Proxy (K1 c f) -> (Maybe Int, Int)

(KnownSymbol field, KnownSymbol field_) => GFieldId field (S1 ('MetaSel ('Just field_) p f b) (Rec0 a))

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy (S1 ('MetaSel ('Just field_) p f b) (Rec0 a)) -> Maybe Int

gfieldId' :: Proxy (S1 ('MetaSel ('Just field_) p f b) (Rec0 a)) -> (Maybe Int, Int)

GFieldId field f => GFieldId field (M1 C t f)

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy (M1 C t f) -> Maybe Int

gfieldId' :: Proxy (M1 C t f) -> (Maybe Int, Int)

GFieldId field f => GFieldId field (M1 D t f)

Defined in Torch.Typed.Lens

Methods

gfieldId :: Proxy (M1 D t f) -> Maybe Int

gfieldId' :: Proxy (M1 D t f) -> (Maybe Int, Int)

Orphan instances

TensorIndex [Maybe Int]

Methods

pushIndex :: [RawTensorIndex] -> [Maybe Int] -> [RawTensorIndex]

toLens :: TensorLike b => [Maybe Int] -> Lens' Tensor b