|
NEML2 2.0.0
|
This is the complete list of members for TensorFromTorchScript, including all inherited members.
| _batch_sizes | TensorBase< Tensor > | protected |
| base_dim() const | TensorBase< Tensor > | |
| base_expand(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
| base_expand(Size base_size, Size dim) const | TensorBase< Tensor > | |
| base_expand_as(const neml2::Tensor &other) const | TensorBase< Tensor > | |
| base_expand_copy(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
| base_flatten() const | TensorBase< Tensor > | |
| base_index(indexing::TensorIndicesRef indices) const | TensorBase< Tensor > | |
| base_index_put_(indexing::TensorIndicesRef indices, const ATensor &other) | TensorBase< Tensor > | |
| base_index_put_(indexing::TensorIndicesRef indices, const CScalar &v) | TensorBase< Tensor > | |
| base_reshape(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
| base_size(Size index) const | TensorBase< Tensor > | |
| base_sizes() const | TensorBase< Tensor > | |
| base_slice(Size dim, const indexing::Slice &index) const | TensorBase< Tensor > | |
| base_storage() const | TensorBase< Tensor > | |
| base_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
| base_unsqueeze(Size d) const | TensorBase< Tensor > | |
| base_unsqueeze_to(Size n) const | Tensor | |
| batch_dim() const | TensorBase< Tensor > | |
| batch_expand(const TraceableTensorShape &batch_shape) const | TensorBase< Tensor > | |
| batch_expand(const TraceableSize &batch_size, Size dim) const | TensorBase< Tensor > | |
| batch_expand_as(const neml2::Tensor &other) const | TensorBase< Tensor > | |
| batch_expand_copy(const TraceableTensorShape &batch_shape) const | TensorBase< Tensor > | |
| batch_index(indexing::TensorIndicesRef indices) const | TensorBase< Tensor > | |
| batch_index_put_(indexing::TensorIndicesRef indices, const ATensor &other) | TensorBase< Tensor > | |
| batch_index_put_(indexing::TensorIndicesRef indices, const CScalar &v) | TensorBase< Tensor > | |
| batch_reshape(const TraceableTensorShape &batch_shape) const | TensorBase< Tensor > | |
| batch_size(Size index) const | TensorBase< Tensor > | |
| batch_sizes() const | TensorBase< Tensor > | |
| batch_slice(Size dim, const indexing::Slice &index) const | TensorBase< Tensor > | |
| batch_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
| batch_unsqueeze(Size d) const | TensorBase< Tensor > | |
| batched() const | TensorBase< Tensor > | |
| clone() const | TensorBase< Tensor > | |
| create(const TensorDataContainer &data, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| create(const TensorDataContainer &data, Size batch_dim, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| detach() const | TensorBase< Tensor > | |
| doc() const | NEML2Object | inline |
| empty(TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| empty(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| empty_like(const Tensor &other) | TensorBase< Tensor > | static |
| expected_options() | TensorFromTorchScript | static |
| factory() const | NEML2Object | inline |
| FromTorchScript(const OptionSet &options) | FromTorchScript | |
| full(TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| full(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const CScalar &init, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| full_like(const Tensor &other, const CScalar &init) | TensorBase< Tensor > | static |
| get_data(const std::string &name) | NEML2Object | |
| get_driver(const std::string &name) | NEML2Object | |
| get_model(const std::string &name) | NEML2Object | |
| get_object(const std::string §ion, const std::string &name) | NEML2Object | |
| get_scheduler(const std::string &name) | NEML2Object | |
| get_solver(const std::string &name) | NEML2Object | |
| host() const | NEML2Object | |
| host() | NEML2Object | |
| identity(Size n, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| identity(const TraceableTensorShape &batch_shape, Size n, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| input_options() const | NEML2Object | inline |
| linspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0) | TensorBase< Tensor > | static |
| load_torch_tensor(const OptionSet &options) const | FromTorchScript | protected |
| logspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, const CScalar &base=10) | TensorBase< Tensor > | static |
| name() const | NEML2Object | inline |
| NEML2Object()=delete | NEML2Object | |
| NEML2Object(NEML2Object &&)=delete | NEML2Object | |
| NEML2Object(const NEML2Object &)=delete | NEML2Object | |
| NEML2Object(const OptionSet &options) | NEML2Object | |
| ones(TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| ones(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| ones_like(const Tensor &other) | TensorBase< Tensor > | static |
| operator-() const | TensorBase< Tensor > | |
| operator=(NEML2Object &&)=delete | NEML2Object | |
| operator=(const NEML2Object &)=delete | NEML2Object | |
| path() const | NEML2Object | inline |
| resolve_tensor(const std::string &name) | NEML2Object | |
| settings() const | NEML2Object | inline |
| setup() | NEML2Object | inlinevirtual |
| Tensor()=default | Tensor | |
| Tensor(const ATensor &tensor, Size batch_dim) | Tensor | |
| Tensor(const ATensor &tensor, const TraceableTensorShape &batch_shape) | Tensor | |
| Tensor(const TensorBase< Derived > &tensor) | Tensor | inline |
| TensorBase()=default | TensorBase< Tensor > | |
| TensorBase(const ATensor &tensor, Size batch_dim) | TensorBase< Tensor > | |
| TensorBase(const ATensor &tensor, const TraceableTensorShape &batch_shape) | TensorBase< Tensor > | |
| TensorBase(const neml2::Tensor &tensor) | TensorBase< Tensor > | |
| TensorBase(double)=delete | TensorBase< Tensor > | |
| TensorBase(float)=delete | TensorBase< Tensor > | |
| TensorBase(int)=delete | TensorBase< Tensor > | |
| TensorFromTorchScript(const OptionSet &options) | TensorFromTorchScript | |
| to(const TensorOptions &options) const | TensorBase< Tensor > | |
| type() const | NEML2Object | inline |
| UserTensorBase(const OptionSet &options) | UserTensorBase | |
| variable_data() const | TensorBase< Tensor > | |
| zeros(TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| zeros(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const TensorOptions &options=default_tensor_options()) | Tensor | static |
| zeros_like(const Tensor &other) | TensorBase< Tensor > | static |
| ~NEML2Object()=default | NEML2Object | virtual |