lamppp
Loading...
Searching...
No Matches
lmp::tensor::TensorImpl Member List

This is the complete list of members for lmp::tensor::TensorImpl, including all inherited members.

copy(const TensorImpl &other) const (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
data() const noexceptlmp::tensor::TensorImpl
device() const noexcept (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
expand_dims(size_t dim) (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
fill(Scalar item) const (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
index(const std::vector< size_t > &idx) (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
numel() const noexcept (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
print(std::ostream &os) const (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
reshape(std::vector< size_t > new_shape) (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
shape() const noexcept (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
squeeze(size_t dim) (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
strides() const noexcept (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl
Tensor (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImplfriend
TensorImpl(const std::vector< T > &data, const std::vector< size_t > &shape, DeviceType device, DataType dtype)lmp::tensor::TensorImplinlineexplicit
TensorImpl(Storage storage, const std::vector< size_t > &shape, DataType dtype) (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImplexplicit
type() const noexcept (defined in lmp::tensor::TensorImpl)lmp::tensor::TensorImpl