Torch.Typed.VLTensor

Documentation

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

A variable length tensor. The length cannot be determined in advance.

Constructors

KnownNat n => VLTensor (Tensor device dtype (n ': shape))

Instances

Show (VLTensor device dtype shape)

Defined in Torch.Typed.VLTensor

Methods

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

show :: VLTensor device dtype shape -> String

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

fromVLTensor :: forall (n :: Nat) (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). (KnownNat n, TensorOptions shape dtype device, KnownShape shape) => VLTensor device dtype shape -> Maybe (Tensor device dtype (n ': shape))

selectIndexes :: forall (n :: Nat) (device :: (DeviceType, Nat)) (dtype :: DType) (shape :: [Nat]). Tensor device dtype (n ': shape) -> Tensor device 'Bool '[n] -> VLTensor device dtype shape

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

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