Read the tensors provided by a torch script file. More...

Detailed Description

Read the tensors provided by a torch script file.

Inheritance diagram for FromTorchScript:

[legend]

Public Member Functions
	FromTorchScript (const OptionSet &options)

Public Member Functions inherited from UserTensorBase
	UserTensorBase (const OptionSet &options)

Public Member Functions inherited from NEML2Object
	NEML2Object ()=delete

	NEML2Object (NEML2Object &&)=delete

	NEML2Object (const NEML2Object &)=delete

NEML2Object &	operator= (NEML2Object &&)=delete

NEML2Object &	operator= (const NEML2Object &)=delete

virtual	~NEML2Object ()=default

	NEML2Object (const OptionSet &options)
	Construct a new NEML2Object object.

const OptionSet &	input_options () const

virtual void	setup ()
	Setup this object.

const std::string &	name () const
	A readonly reference to the object's name.

const std::string &	type () const
	A readonly reference to the object's type.

const std::string &	path () const
	A readonly reference to the object's path.

const std::string &	doc () const
	A readonly reference to the object's docstring.

template<typename T = NEML2Object>
const T *	host () const
	Get a readonly pointer to the host.

template<typename T = NEML2Object>
T *	host ()
	Get a writable pointer to the host.

Static Public Member Functions
static OptionSet	expected_options ()

Static Public Member Functions inherited from UserTensorBase
static OptionSet	expected_options ()

Static Public Member Functions inherited from NEML2Object
static OptionSet	expected_options ()

Protected Member Functions
at::Tensor	load_torch_tensor (const OptionSet &options) const
	Helper to make the tensor given user input options.

Constructor & Destructor Documentation

FromTorchScript ( const OptionSet & options )

OptionSet expected_options ( )

static

torch::Tensor load_torch_tensor ( const OptionSet & options ) const

protected

Helper to make the tensor given user input options.