Sha256: 47fb719d9d96cf7887905555b8935f3d2390f3dc852ddc5cc2a41c6ee3a378f7
Contents?: true
Size: 549 Bytes
Versions: 1
Compression:
Stored size: 549 Bytes
Contents
#include <torch/torch.h> #include <rice/Module.hpp> #include "utils.h" void init_random(Rice::Module& m) { Rice::define_module_under(m, "Random") .add_handler<torch::Error>(handle_error) .define_singleton_method( "initial_seed", *[]() { return at::detail::getDefaultCPUGenerator().current_seed(); }) .define_singleton_method( "seed", *[]() { // TODO set for CUDA when available auto generator = at::detail::getDefaultCPUGenerator(); return generator.seed(); }); }
Version data entries
1 entries across 1 versions & 1 rubygems
Version | Path |
---|---|
torch-rb-0.6.0 | ext/torch/random.cpp |