NEML2 1.4.0
Loading...
Searching...
No Matches
UserTensor Member List

This is the complete list of members for UserTensor, including all inherited members.

base_dim() constTensorBase< Tensor >
base_expand(TensorShapeRef base_size) constTensorBase< Tensor >
base_expand_as(const Derived2 &other) constTensorBase< Tensor >
base_expand_copy(TensorShapeRef base_size) constTensorBase< Tensor >
base_index(indexing::TensorIndicesRef indices) constTensorBase< Tensor >
base_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other)TensorBase< Tensor >
base_reshape(TensorShapeRef base_shape) constTensorBase< Tensor >
base_size(Size index) constTensorBase< Tensor >
base_sizes() constTensorBase< Tensor >
base_storage() constTensorBase< Tensor >
base_transpose(Size d1, Size d2) constTensorBase< Tensor >
base_unsqueeze(Size d) constTensorBase< Tensor >
batch_dim() constTensorBase< Tensor >
batch_expand(TensorShapeRef batch_size) constTensorBase< Tensor >
batch_expand_as(const Derived2 &other) constTensorBase< Tensor >
batch_expand_copy(TensorShapeRef batch_size) constTensorBase< Tensor >
batch_index(indexing::TensorIndicesRef indices) constTensorBase< Tensor >
batch_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other)TensorBase< Tensor >
batch_reshape(TensorShapeRef batch_shape) constTensorBase< Tensor >
batch_size(Size index) constTensorBase< Tensor >
batch_sizes() constTensorBase< Tensor >
batch_transpose(Size d1, Size d2) constTensorBase< Tensor >
batch_unsqueeze(Size d) constTensorBase< Tensor >
batched() constTensorBase< Tensor >
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) constTensorBase< Tensor >
detach() constTensorBase< Tensor >
doc() constNEML2Objectinline
empty(const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
empty(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
empty_like(const Tensor &other)TensorBase< Tensor >static
expected_options()UserTensorstatic
full(const TensorShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
full(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
full_like(const Tensor &other, Real init)TensorBase< Tensor >static
host() constNEML2Object
host()NEML2Object
identity(Size n, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
identity(const TensorShapeRef &batch_shape, Size n, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
input_options() constNEML2Objectinline
linspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, Size batch_dim=-1)TensorBase< Tensor >static
logspace(const Tensor &start, const Tensor &end, Size nstep, Size dim=0, Size batch_dim=-1, Real base=10)TensorBase< Tensor >static
name() constNEML2Objectinline
NEML2Object(const OptionSet &options)NEML2Object
ones(const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
ones(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
ones_like(const Tensor &other)TensorBase< Tensor >static
operator-() constTensorBase< Tensor >
path() constNEML2Objectinline
setup()NEML2Objectinlinevirtual
TensorBase()=defaultTensorBase< Tensor >
TensorBase(const torch::Tensor &tensor, Size batch_dim)TensorBase< Tensor >
TensorBase(const Tensor &tensor)TensorBase< Tensor >
TensorBase(Real)=deleteTensorBase< Tensor >
to(const torch::TensorOptions &options) constTensorBase< Tensor >
type() constNEML2Objectinline
UserTensor(const OptionSet &options)UserTensor
UserTensorBase(const OptionSet &options)UserTensorBase
zeros(const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
zeros(const TensorShapeRef &batch_shape, const TensorShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())Tensorstatic
zeros_like(const Tensor &other)TensorBase< Tensor >static
~NEML2Object()=defaultNEML2Objectvirtual