Documentation

Swaps the arguments of HaskTorch's foldLoop around

Because snake_case sucks

Default limits for nanToNum

Default limits for nanToNum (0.0)

GPU Tensor filled with Float value

Select index with [Int] from GPU tensor

Torch.where' with fixed type for where'

Syntactic sugar for HaskTorch's repeatInterleave so it can more easily be fmapped.

Helper function creating split indices as gpu int tensor splits' :: [Int] -> [T.Tensor] splits' = map tit . splits where tit i = T.asTensor (i :: Int)

Split Tensor into list of Tensors along dimension

Create Boolean Mask Tensor from list of indices.

Create a Boolean Mask Tensor from index Tensor

Manually set RNG Seed

Get the current RNG Seed

GPU 1

CPU 0

Default Tensor Data Type

Convert an Array to a Tensor on GPU

Convert an Array to a Tensor on CPU

Convert an Array to a Tensor on GPU

Convert an Array to a Tensor on CPU

Create an empty Float Tensor on GPU

Create an empty Float Tensor on CPU

Convert a Scalar to a Tensor on GPU

Convert a Scalar to a Tensor on CPU

Convert model to Double on GPU

Convert model to Double on CPU

Convert model to Float on CPU

Convert model to Float on CPU

Generate a Tensor of random Integers on GPU

Generate a Tensor of random Integers on CPU

Generate Normally Distributed Random values given dimensions

Generate Uniformally distributed values in a given range

Rescale tensor s.t. mean = 0.0 and std = 1.0