doxygen/moose/LibtorchRandomUtils_8h_source.html

//* This file is part of the MOOSE framework

//* https://mooseframework.inl.gov

//*

//* All rights reserved, see COPYRIGHT for full restrictions

//* https://github.com/idaholab/moose/blob/master/COPYRIGHT

//*

//* Licensed under LGPL 2.1, please see LICENSE for details

//* https://www.gnu.org/licenses/lgpl-2.1.html


#ifdef MOOSE_LIBTORCH_ENABLED


#pragma once


#include <torch/torch.h>


#include "MooseTypes.h"


namespace Moose

{


at::Generator makeLibtorchCPUGenerator();


at::Generator makeLibtorchCPUGenerator(uint64_t seed);


void orthogonalInitializeTensor(torch::Tensor & tensor,

                                Real gain = 1.0,

                                c10::optional<at::Generator> generator = c10::nullopt);


} // namespace Moose


#endif

MooseTypes.h

Moose
MOOSE now contains C++17 code, so give a reasonable error message stating what the user can do to add...
Definition SetAdaptivityOptionsAction.h:15

Moose::orthogonalInitializeTensor
void orthogonalInitializeTensor(torch::Tensor &tensor, Real gain=1.0, c10::optional< at::Generator > generator=c10::nullopt)
Fill a tensor with a (semi) orthogonal matrix using the provided generator.
Definition LibtorchRandomUtils.C:36

Moose::makeLibtorchCPUGenerator
at::Generator makeLibtorchCPUGenerator()
Create an owned CPU generator using libtorch's default seed behavior.
Definition LibtorchRandomUtils.C:24