NEML2 2.0.0
Loading...
Searching...
No Matches
Vec Member List

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

base_dim() constTensorBase< Derived >
base_expand(TensorShapeRef base_shape) constTensorBase< Derived >
base_expand(Size base_size, Size dim) constTensorBase< Derived >
base_expand_as(const Derived2 &other) constTensorBase< Derived >
base_expand_copy(TensorShapeRef base_shape) constTensorBase< Derived >
base_flatten() constTensorBase< Derived >
base_index(indexing::TensorIndicesRef indices) constTensorBase< Derived >
base_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other)TensorBase< Derived >
base_index_put_(indexing::TensorIndicesRef indices, Real v)TensorBase< Derived >
base_reshape(TensorShapeRef base_shape) constTensorBase< Derived >
base_size(Size index) constTensorBase< Derived >
base_sizes() constTensorBase< Derived >
base_storage() constTensorBase< Derived >
base_transpose(Size d1, Size d2) constTensorBase< Derived >
base_unsqueeze(Size d) constTensorBase< Derived >
batch_dim() constTensorBase< Derived >
batch_expand(const TraceableTensorShape &batch_shape) constTensorBase< Derived >
batch_expand(const TraceableSize &batch_size, Size dim) constTensorBase< Derived >
batch_expand_as(const Derived2 &other) constTensorBase< Derived >
batch_expand_copy(const TraceableTensorShape &batch_shape) constTensorBase< Derived >
batch_index(indexing::TensorIndicesRef indices) constTensorBase< Derived >
batch_index_put_(indexing::TensorIndicesRef indices, const torch::Tensor &other)TensorBase< Derived >
batch_index_put_(indexing::TensorIndicesRef indices, Real v)TensorBase< Derived >
batch_reshape(const TraceableTensorShape &batch_shape) constTensorBase< Derived >
batch_size(Size index) constTensorBase< Derived >
batch_sizes() constTensorBase< Derived >
batch_transpose(Size d1, Size d2) constTensorBase< Derived >
batch_unsqueeze(Size d) constTensorBase< Derived >
batched() constTensorBase< Derived >
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) constTensorBase< Derived >
const_base_dimPrimitiveTensor< Derived, S >static
const_base_sizesPrimitiveTensor< Derived, S >inlinestatic
const_base_storagePrimitiveTensor< Derived, S >inlinestatic
cross(const VecBase< Derived2 > &v) constVecBase< Vec >
detach() constTensorBase< Derived >
dot(const VecBase< Derived2 > &v) constVecBase< Vec >
drotate(const Rot &r) constVecBase< Vec >
drotate(const R2 &R) constVecBase< Vec >
empty(const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
empty(const TraceableTensorShape &batch_shape, const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
empty_like(const Derived &other)TensorBase< Derived >static
fill(const Real &v1, const Real &v2, const Real &v3, const torch::TensorOptions &options=default_tensor_options())VecBase< Vec >static
fill(const Scalar &v1, const Scalar &v2, const Scalar &v3)VecBase< Vec >static
full(Real init, const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
full(const TraceableTensorShape &batch_shape, Real init, const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
full_like(const Derived &other, Real init)TensorBase< Derived >static
identity_map(const torch::TensorOptions &options=default_tensor_options())Vecstatic
linspace(const Derived &start, const Derived &end, Size nstep, Size dim=0)TensorBase< Derived >static
logspace(const Derived &start, const Derived &end, Size nstep, Size dim=0, Real base=10)TensorBase< Derived >static
norm() constVecBase< Vec >
norm_sq() constVecBase< Vec >
ones(const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
ones(const TraceableTensorShape &batch_shape, const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
ones_like(const Derived &other)TensorBase< Derived >static
operator Tensor() constPrimitiveTensor< Derived, S >
operator()(Size i) constVecBase< Vec >
operator-() constTensorBase< Derived >
operator=(const Transformable &)=defaultTransformable< Vec >
operator=(Transformable &&) noexcept=defaultTransformable< Vec >
outer(const VecBase< Derived2 > &v) constVecBase< Vec >
PrimitiveTensor()=defaultPrimitiveTensor< Derived, S >
PrimitiveTensor(const torch::Tensor &tensor, Size batch_dim)PrimitiveTensor< Derived, S >explicit
PrimitiveTensor(const torch::Tensor &tensor, const TraceableTensorShape &batch_shape)PrimitiveTensor< Derived, S >explicit
PrimitiveTensor(const TensorBase< Derived2 > &tensor)PrimitiveTensor< Derived, S >
PrimitiveTensor(const torch::Tensor &tensor)PrimitiveTensor< Derived, S >
rotate(const Rot &r) constVecBase< Vec >
rotate(const R2 &R) constVecBase< Vec >
TensorBase()=defaultTensorBase< Derived >
TensorBase(const torch::Tensor &tensor, Size batch_dim)TensorBase< Derived >
TensorBase(const torch::Tensor &tensor, const TraceableTensorShape &batch_shape)TensorBase< Derived >
TensorBase(const TensorBase< Derived2 > &tensor)TensorBase< Derived >
TensorBase(Real)=deleteTensorBase< Derived >
to(const torch::TensorOptions &options) constTensorBase< Derived >
transform(const R2 &op) const overrideVecvirtual
Transformable()=defaultTransformable< Vec >
Transformable(const Transformable &)=defaultTransformable< Vec >
Transformable(Transformable &&) noexcept=defaultTransformable< Vec >
variable_data() constTensorBase< Derived >
Vec(const Rot &r)Vec
zeros(const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
zeros(const TraceableTensorShape &batch_shape, const torch::TensorOptions &options=default_tensor_options())PrimitiveTensor< Derived, S >static
zeros_like(const Derived &other)TensorBase< Derived >static
~Transformable()=defaultTransformable< Vec >virtual