索引

A

abs() （torch_sla.SparseTensor 方法）

abs() （torch_sla.SparseTensorList 方法） adjoint_solve()（在 torch_sla 模块中）

B

batch_shape（torch_sla.SparseTensor 属性）, [1] batch_size（torch_sla.SparseTensor 属性）

block_shape（torch_sla.SparseTensor 属性）, [1] block_sizes（torch_sla.SparseTensorList 属性）

C

ceil() （torch_sla.SparseTensor 方法） clamp() （torch_sla.SparseTensor 方法） clamp() （torch_sla.SparseTensorList 方法） clone() （torch_sla.SparseTensor 方法） col_indices（torch_sla.SparseTensor 属性） condition_number() （torch_sla.DSparseTensor 方法） condition_number() （torch_sla.SparseTensor 方法） condition_number() （torch_sla.SparseTensorList 方法） connected_components() （torch_sla.SparseTensor 方法） contiguous() （torch_sla.SparseTensor 方法）

cos() （torch_sla.SparseTensor 方法） cosh() （torch_sla.SparseTensor 方法） cpu() （torch_sla.DSparseMatrix 方法） cpu() （torch_sla.DSparseTensor 方法） cpu() （torch_sla.SparseTensor 方法） cpu() （torch_sla.SparseTensorList 方法） cuda() （torch_sla.DSparseMatrix 方法） cuda() （torch_sla.DSparseTensor 方法） cuda() （torch_sla.SparseTensor 方法） cuda() （torch_sla.SparseTensorList 方法）

D

det() （torch_sla.DSparseMatrix 方法） det() （torch_sla.DSparseTensor 方法） det() （torch_sla.SparseTensor 方法） det() （torch_sla.SparseTensorList 方法） detach() （torch_sla.SparseTensor 方法） device（torch_sla.DSparseMatrix 属性） device（torch_sla.DSparseTensor 属性） device（torch_sla.SparseTensor 属性）

device（torch_sla.SparseTensorList 属性）, [1] double() （torch_sla.SparseTensor 方法） DSparseMatrix（torch_sla 中的类） DSparseTensor（torch_sla 中的类） dtype（torch_sla.DSparseMatrix 属性） dtype（torch_sla.DSparseTensor 属性） dtype（torch_sla.SparseTensor 属性） dtype（torch_sla.SparseTensorList 属性）, [1]

E

eigs() （torch_sla.DSparseTensor 方法） eigs() （torch_sla.SparseTensor 方法） eigs() （torch_sla.SparseTensorList 方法） eigsh() （torch_sla.DSparseMatrix 方法）

eigsh() （torch_sla.DSparseTensor 方法） eigsh() （torch_sla.SparseTensor 方法） eigsh() （torch_sla.SparseTensorList 方法） exp() （torch_sla.SparseTensor 方法） exp() （torch_sla.SparseTensorList 方法）

F

flatten_blocks() （torch_sla.SparseTensor 方法） float() （torch_sla.SparseTensor 方法） floor() （torch_sla.SparseTensor 方法） from_block_diagonal()（torch_sla.SparseTensorList 类方法） from_coo_list()（torch_sla.SparseTensorList 类方法） from_dense()（torch_sla.SparseTensor 类方法） from_device_mesh()（torch_sla.DSparseTensor 类方法）

from_global()（torch_sla.DSparseMatrix 类方法） from_global_distributed()（torch_sla.DSparseTensor 类方法） from_list()（torch_sla.DSparseTensor 类方法） from_sparse_tensor()（torch_sla.DSparseTensor 类方法） from_torch_sparse()（torch_sla.DSparseTensor 类方法） from_torch_sparse()（torch_sla.SparseTensor 类方法） from_torch_sparse_list()（torch_sla.SparseTensorList 类方法）

G

gather() （torch_sla.DSparseTensor 方法） gather_from_dtensor() （torch_sla.DSparseTensor 方法） gather_global() （torch_sla.DSparseMatrix 方法） gather_global() （torch_sla.DSparseTensor 方法）

get_available_backends()（在 torch_sla 模块中） get_backend_methods()（在 torch_sla 模块中） global_shape（torch_sla.DSparseMatrix 属性） grad（torch_sla.SparseTensor 属性）

H

half() （torch_sla.SparseTensor 方法） halo_exchange() （torch_sla.DSparseMatrix 方法） halo_exchange_async() （torch_sla.DSparseMatrix 方法）

halo_exchange_local() （torch_sla.DSparseMatrix 方法） halo_exchange_local() （torch_sla.DSparseTensor 方法） has_isolated_components() （torch_sla.DSparseTensor 方法） has_isolated_components() （torch_sla.SparseTensor 方法）

I

is_batched（torch_sla.SparseTensor 属性） is_block（torch_sla.SparseTensor 属性） is_cuda（torch_sla.DSparseMatrix 属性） is_cuda（torch_sla.DSparseTensor 属性） is_cuda（torch_sla.SparseTensor 属性） is_positive_definite() （torch_sla.DSparseTensor 方法） is_positive_definite() （torch_sla.SparseTensor 方法）

is_positive_definite() （torch_sla.SparseTensorList 方法） is_square（torch_sla.SparseTensor 属性） is_symmetric() （torch_sla.DSparseTensor 方法） is_symmetric() （torch_sla.SparseTensor 方法） is_symmetric() （torch_sla.SparseTensorList 方法） isfinite() （torch_sla.SparseTensor 方法） isinf() （torch_sla.SparseTensor 方法） isnan() （torch_sla.SparseTensor 方法）

L

load()（torch_sla.DSparseMatrix 类方法） load()（torch_sla.DSparseTensor 类方法） load()（torch_sla.SparseTensor 类方法） load_mtx()（在 torch_sla 模块中） load_partition()（在 torch_sla 模块中） load_sparse()（在 torch_sla 模块中） local_col（torch_sla.DSparseMatrix 属性） local_row（torch_sla.DSparseMatrix 属性） local_shape（torch_sla.DSparseMatrix 属性） local_values（torch_sla.DSparseMatrix 属性）

log() （torch_sla.SparseTensor 方法） log() （torch_sla.SparseTensorList 方法） log10() （torch_sla.SparseTensor 方法） log2() （torch_sla.SparseTensor 方法） logical_and() （torch_sla.SparseTensor 方法） logical_not() （torch_sla.SparseTensor 方法） logical_or() （torch_sla.SparseTensor 方法） logical_xor() （torch_sla.SparseTensor 方法） lu() （torch_sla.DSparseTensor 方法） lu() （torch_sla.SparseTensor 方法） LUFactorization（torch_sla 中的类）

M

matvec() （torch_sla.DSparseMatrix 方法） matvec_all() （torch_sla.DSparseTensor 方法） matvec_overlap() （torch_sla.DSparseMatrix 方法） max() （torch_sla.SparseTensor 方法）

max() （torch_sla.SparseTensorList 方法） mean() （torch_sla.SparseTensor 方法） mean() （torch_sla.SparseTensorList 方法） min() （torch_sla.SparseTensor 方法） min() （torch_sla.SparseTensorList 方法）

N

ndim（torch_sla.SparseTensor 属性） nnz（torch_sla.DSparseMatrix 属性） nnz（torch_sla.DSparseTensor 属性） nnz（torch_sla.SparseTensor 属性） nonlinear_solve() （torch_sla.DSparseTensor 方法） nonlinear_solve() （torch_sla.SparseTensor 方法） nonlinear_solve()（在 torch_sla 模块中）

norm() （torch_sla.DSparseTensor 方法） norm() （torch_sla.SparseTensor 方法） norm() （torch_sla.SparseTensorList 方法） num_halo（torch_sla.DSparseMatrix 属性） num_local（torch_sla.DSparseMatrix 属性） num_owned（torch_sla.DSparseMatrix 属性） num_partitions（torch_sla.DSparseTensor 属性）

P

partition() （torch_sla.SparseTensor 方法） partition() （torch_sla.SparseTensorList 方法） partition_for_rank() （torch_sla.SparseTensor 方法） partition_ids（torch_sla.DSparseTensor 属性）

partition（torch_sla.DSparseMatrix 属性） pow() （torch_sla.SparseTensor 方法） pow() （torch_sla.SparseTensorList 方法） prod() （torch_sla.SparseTensor 方法）

R

reciprocal() （torch_sla.SparseTensor 方法） relu() （torch_sla.SparseTensor 方法） requires_grad_() （torch_sla.SparseTensor 方法）

requires_grad（torch_sla.SparseTensor 属性） round() （torch_sla.SparseTensor 方法） row_indices（torch_sla.SparseTensor 属性）

S

save() （torch_sla.DSparseTensor 方法） save() （torch_sla.SparseTensor 方法） save_distributed() （torch_sla.SparseTensor 方法） save_distributed()（在 torch_sla 模块中） save_mtx()（在 torch_sla 模块中） save_sparse()（在 torch_sla 模块中） scatter_local() （torch_sla.DSparseTensor 方法） scatter_to_dtensor() （torch_sla.DSparseTensor 方法） select_backend()（在 torch_sla 模块中） select_method()（在 torch_sla 模块中） shapes（torch_sla.SparseTensorList 属性）, [1] shape（torch_sla.DSparseTensor 属性） shape（torch_sla.SparseTensor 属性）, [1] sigmoid() （torch_sla.SparseTensor 方法） sign() （torch_sla.SparseTensor 方法） sin() （torch_sla.SparseTensor 方法） sinh() （torch_sla.SparseTensor 方法） solve() （torch_sla.DSparseMatrix 方法） solve() （torch_sla.LUFactorization 方法） solve() （torch_sla.SparseTensor 方法） solve() （torch_sla.SparseTensorList 方法） solve_all() （torch_sla.DSparseTensor 方法）

solve_batch() （torch_sla.SparseTensor 方法） solve_distributed() （torch_sla.DSparseTensor 方法） sparse_dim（torch_sla.SparseTensor 属性） sparse_shape（torch_sla.SparseTensor 属性）, [1] SparseTensorList（torch_sla 中的类） SparseTensor（torch_sla 中的类） spsolve()（在 torch_sla 模块中） spsolve_batch_different_layout()（在 torch_sla 模块中） spsolve_batch_same_layout()（在 torch_sla 模块中） spsolve_coo()（在 torch_sla 模块中） spsolve_csr()（在 torch_sla 模块中） spy() （torch_sla.DSparseTensor 方法） spy() （torch_sla.SparseTensor 方法） spy() （torch_sla.SparseTensorList 方法） sqrt() （torch_sla.SparseTensor 方法） sqrt() （torch_sla.SparseTensorList 方法） square() （torch_sla.SparseTensor 方法） sum() （torch_sla.SparseTensor 方法） sum() （torch_sla.SparseTensorList 方法） supports_dtensor（torch_sla.DSparseTensor 属性） svd() （torch_sla.DSparseTensor 方法） svd() （torch_sla.SparseTensor 方法） svd() （torch_sla.SparseTensorList 方法）

T

T() （torch_sla.DSparseTensor 方法） T() （torch_sla.SparseTensor 方法） tan() （torch_sla.SparseTensor 方法） tanh() （torch_sla.SparseTensor 方法） to() （torch_sla.DSparseMatrix 方法） to() （torch_sla.DSparseTensor 方法） to() （torch_sla.SparseTensor 方法） to() （torch_sla.SparseTensorList 方法） to_block_diagonal() （torch_sla.SparseTensorList 方法）

to_connected_components() （torch_sla.SparseTensor 方法） to_csr() （torch_sla.SparseTensor 方法） to_dense() （torch_sla.DSparseTensor 方法） to_dense() （torch_sla.SparseTensor 方法） to_dtensor() （torch_sla.DSparseTensor 方法） to_list() （torch_sla.DSparseTensor 方法） to_sparse_tensor() （torch_sla.DSparseTensor 方法） to_torch_sparse() （torch_sla.SparseTensor 方法） total_nnz（torch_sla.SparseTensorList 属性） total_shape（torch_sla.SparseTensorList 属性）

U

unflatten_blocks() （torch_sla.SparseTensor 方法）

V

values（torch_sla.SparseTensor 属性）