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

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

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