NEML2 2.0.0
|
This is the complete list of members for OnesTensor, including all inherited members.
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 Derived2 &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 torch::Tensor &other) | TensorBase< Tensor > | |
base_index_put_(indexing::TensorIndicesRef indices, Real v) | TensorBase< Tensor > | |
base_reshape(TensorShapeRef base_shape) const | TensorBase< Tensor > | |
base_size(Size index) const | TensorBase< Tensor > | |
base_sizes() const | TensorBase< Tensor > | |
base_storage() const | TensorBase< Tensor > | |
base_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
base_unsqueeze(Size d) const | TensorBase< 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 Derived2 &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 torch::Tensor &other) | TensorBase< Tensor > | |
batch_index_put_(indexing::TensorIndicesRef indices, Real 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_transpose(Size d1, Size d2) const | TensorBase< Tensor > | |
batch_unsqueeze(Size d) const | TensorBase< Tensor > | |
batched() const | TensorBase< Tensor > | |
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) const | TensorBase< Tensor > | |
detach() const | TensorBase< Tensor > | |
doc() const | NEML2Object | inline |
empty(TensorShapeRef base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
empty(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
empty_like(const Tensor &other) | TensorBase< Tensor > | static |
expected_options() | OnesTensor | static |
full(TensorShapeRef base_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
full(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, Real init, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
full_like(const Tensor &other, Real init) | TensorBase< Tensor > | static |
host() const | NEML2Object | |
host() | NEML2Object | |
identity(Size n, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
identity(const TraceableTensorShape &batch_shape, Size n, const torch::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 |
logspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, Real 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 torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
ones(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
ones_like(const Tensor &other) | TensorBase< Tensor > | static |
OnesTensor(const OptionSet &options) | OnesTensor | |
operator-() const | TensorBase< Tensor > | |
operator=(const NEML2Object &)=delete | NEML2Object | |
operator=(NEML2Object &&)=delete | NEML2Object | |
path() const | NEML2Object | inline |
setup() | NEML2Object | inlinevirtual |
Tensor()=default | Tensor | |
Tensor(const torch::Tensor &tensor, Size batch_dim) | Tensor | |
Tensor(const torch::Tensor &tensor, const TraceableTensorShape &batch_shape) | Tensor | |
Tensor(const TensorBase< Derived2 > &tensor) | Tensor | inline |
TensorBase()=default | TensorBase< Tensor > | |
TensorBase(const torch::Tensor &tensor, Size batch_dim) | TensorBase< Tensor > | |
TensorBase(const torch::Tensor &tensor, const TraceableTensorShape &batch_shape) | TensorBase< Tensor > | |
TensorBase(const TensorBase< Derived2 > &tensor) | TensorBase< Tensor > | |
TensorBase(Real)=delete | TensorBase< Tensor > | |
to(const torch::TensorOptions &options) const | TensorBase< Tensor > | |
type() const | NEML2Object | inline |
UserTensorBase(const OptionSet &options) | UserTensorBase | |
variable_data() const | TensorBase< Tensor > | |
zeros(TensorShapeRef base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
zeros(const TraceableTensorShape &batch_shape, TensorShapeRef base_shape, const torch::TensorOptions &options=default_tensor_options()) | Tensor | static |
zeros_like(const Tensor &other) | TensorBase< Tensor > | static |
~NEML2Object()=default | NEML2Object | virtual |