module

Torch

Class methods

check!

Raises Torch::Error if the shim recorded an error for the last call.

Source
cuda_available?
Source
cuda_device_count
Source
cuda_manual_seed(seed : UInt64) : Nil
Source
cuda_synchronize(device_index : Int32 = 0) : Nil
Source
last_error

Returns and clears the last shim error on this thread (set by PROTECT).

Source
manual_seed(seed : Int64) : Nil
Source
num_threads
Source
num_threads=(n : Int32) : Nil
Source

Nested types