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

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

base_dim() constBatchTensorBase< BatchTensor >
base_expand(TorchShapeRef base_size) constBatchTensorBase< BatchTensor >
base_expand_as(const Derived2 &other) constBatchTensorBase< BatchTensor >
base_expand_copy(TorchShapeRef base_size) constBatchTensorBase< BatchTensor >
base_index(const TorchSlice &indices) constBatchTensorBase< BatchTensor >
base_index_put(const TorchSlice &indices, const torch::Tensor &other)BatchTensorBase< BatchTensor >
base_movedim(TorchSize d1, TorchSize d2) constBatchTensorBase< BatchTensor >
base_reshape(TorchShapeRef base_shape) constBatchTensorBase< BatchTensor >
base_size(TorchSize index) constBatchTensorBase< BatchTensor >
base_sizes() constBatchTensorBase< BatchTensor >
base_storage() constBatchTensorBase< BatchTensor >
base_transpose(TorchSize d1, TorchSize d2) constBatchTensorBase< BatchTensor >
base_unsqueeze(TorchSize d) constBatchTensorBase< BatchTensor >
batch_dim() constBatchTensorBase< BatchTensor >
batch_dim()BatchTensorBase< BatchTensor >
batch_expand(TorchShapeRef batch_size) constBatchTensorBase< BatchTensor >
batch_expand_as(const Derived2 &other) constBatchTensorBase< BatchTensor >
batch_expand_copy(TorchShapeRef batch_size) constBatchTensorBase< BatchTensor >
batch_index(TorchSlice indices) constBatchTensorBase< BatchTensor >
batch_index_put(TorchSlice indices, const torch::Tensor &other)BatchTensorBase< BatchTensor >
batch_reshape(TorchShapeRef batch_shape) constBatchTensorBase< BatchTensor >
batch_size(TorchSize index) constBatchTensorBase< BatchTensor >
batch_sizes() constBatchTensorBase< BatchTensor >
batch_sum(TorchSize d) constBatchTensorBase< BatchTensor >
batch_transpose(TorchSize d1, TorchSize d2) constBatchTensorBase< BatchTensor >
batch_unsqueeze(TorchSize d) constBatchTensorBase< BatchTensor >
batched() constBatchTensorBase< BatchTensor >
BatchTensorBase()=defaultBatchTensorBase< BatchTensor >
BatchTensorBase(const torch::Tensor &tensor, TorchSize batch_dim)BatchTensorBase< BatchTensor >
BatchTensorBase(const BatchTensor &tensor)BatchTensorBase< BatchTensor >
BatchTensorBase(Real)=deleteBatchTensorBase< BatchTensor >
clone(torch::MemoryFormat memory_format=torch::MemoryFormat::Contiguous) constBatchTensorBase< BatchTensor >
detach() constBatchTensorBase< BatchTensor >
empty(const TorchShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
empty(const TorchShapeRef &batch_shape, const TorchShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
empty_like(const BatchTensor &other)BatchTensorBase< BatchTensor >static
full(const TorchShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
full(const TorchShapeRef &batch_shape, const TorchShapeRef &base_shape, Real init, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
full_like(const BatchTensor &other, Real init)BatchTensorBase< BatchTensor >static
identity(TorchSize n, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
identity(const TorchShapeRef &batch_shape, TorchSize n, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
linspace(const BatchTensor &start, const BatchTensor &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1)BatchTensorBase< BatchTensor >static
list_sum() constBatchTensorBase< BatchTensor >
list_unsqueeze() constBatchTensorBase< BatchTensor >
logspace(const BatchTensor &start, const BatchTensor &end, TorchSize nstep, TorchSize dim=0, TorchSize batch_dim=-1, Real base=10)BatchTensorBase< BatchTensor >static
ones(const TorchShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
ones(const TorchShapeRef &batch_shape, const TorchShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
ones_like(const BatchTensor &other)BatchTensorBase< BatchTensor >static
operator-() constBatchTensorBase< BatchTensor >
to(const torch::TensorOptions &options) constBatchTensorBase< BatchTensor >
zeros(const TorchShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
zeros(const TorchShapeRef &batch_shape, const TorchShapeRef &base_shape, const torch::TensorOptions &options=default_tensor_options())BatchTensorstatic
zeros_like(const BatchTensor &other)BatchTensorBase< BatchTensor >static