libtorch-ffi-1.11.0.0: test out alternative options for ffi interface to libtorch 1.x
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.Internal.Managed.Type.TensorIndex

Documentation

newTensorIndexList :: IO (ForeignPtr (StdVector TensorIndex)) Source #

newTensorIndexWithInt :: CInt -> IO (ForeignPtr TensorIndex) Source #

newTensorIndexWithBool :: CBool -> IO (ForeignPtr TensorIndex) Source #

newTensorIndexWithSlice :: CInt -> CInt -> CInt -> IO (ForeignPtr TensorIndex) Source #

newTensorIndexWithTensor :: ForeignPtr Tensor -> IO (ForeignPtr TensorIndex) Source #

newTensorIndexWithEllipsis :: IO (ForeignPtr TensorIndex) Source #

newTensorIndexWithNone :: IO (ForeignPtr TensorIndex) Source #

tensorIndexList_empty :: ForeignPtr (StdVector TensorIndex) -> IO CBool Source #

tensorIndexList_size :: ForeignPtr (StdVector TensorIndex) -> IO CSize Source #

tensorIndexList_push_back :: ForeignPtr (StdVector TensorIndex) -> ForeignPtr TensorIndex -> IO () Source #

index :: ForeignPtr Tensor -> ForeignPtr (StdVector TensorIndex) -> IO (ForeignPtr Tensor) Source #

index_put_ :: ForeignPtr Tensor -> ForeignPtr (StdVector TensorIndex) -> ForeignPtr Tensor -> IO (ForeignPtr Tensor) Source #