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