Torch.Indef.Static.Tensor.Index

Documentation

_indexCopy :: Tensor d -> Int -> IndexTensor '[n] -> Tensor d' -> IO ()

Static call to _indexCopy

_indexAdd :: Tensor d -> Int -> IndexTensor '[n] -> Tensor d' -> IO ()

Static call to _indexAdd

_indexFill :: Tensor d -> Int -> IndexTensor '[n] -> HsReal -> IO ()

Static call to _indexFill

_indexSelect :: Tensor d -> Tensor d' -> Int -> IndexTensor '[n] -> IO ()

Static call to _indexSelect

_take :: Tensor d -> Tensor d' -> IndexTensor '[n] -> IO ()

Static call to _take

_put :: Tensor d -> IndexTensor '[n] -> Tensor d' -> Int -> IO ()

Static call to _put

getRow :: forall t n m. (All KnownDim '[n, m], KnownNat m) => Tensor '[n, m] -> Word -> Maybe (Tensor '[1, m])

Retrieve a single row from a matrix

FIXME: Use Idx and remove the throwString function

getColumn :: forall t n m. (All KnownDim '[n, m], KnownNat n) => Tensor '[n, m] -> Word -> Maybe (Tensor '[n, 1])

Retrieve a single column from a matrix

FIXME: Use Idx and remove the throwString function