Index

A

AbstractLayer (class in nngeometry.layercollection) add_layer() (nngeometry.layercollection.LayerCollection method)

add_layer_from_model() (nngeometry.layercollection.LayerCollection method) add_to_model() (nngeometry.object.vector.PVector method) Affine1dLayer (class in nngeometry.layercollection)

B

BatchNorm1dLayer (class in nngeometry.layercollection)

BatchNorm2dLayer (class in nngeometry.layercollection) bdot() (in module nngeometry.object.pspace)

C

clone() (nngeometry.object.vector.PVector method) Conv1dLayer (class in nngeometry.layercollection) Conv2dLayer (class in nngeometry.layercollection)

ConvTranspose2dLayer (class in nngeometry.layercollection) copy_to_model() (nngeometry.object.vector.PVector method) Cosine1dLayer (class in nngeometry.layercollection)

D

detach() (nngeometry.object.vector.PVector method)

dot() (nngeometry.object.vector.PVector method)

E

EmbeddingLayer (class in nngeometry.layercollection)

F

from_model() (nngeometry.layercollection.LayerCollection method) (nngeometry.object.vector.PVector static method)

from_model_grad() (nngeometry.object.vector.PVector static method) FVector (class in nngeometry.object.vector)

G

get_common_layers() (nngeometry.layercollection.LayerCollection method) get_diag() (nngeometry.object.pspace.PMatAbstract method) (nngeometry.object.pspace.PMatBlockDiag method) (nngeometry.object.pspace.PMatDense method) (nngeometry.object.pspace.PMatDiag method) (nngeometry.object.pspace.PMatEKFAC method) (nngeometry.object.pspace.PMatEye method) (nngeometry.object.pspace.PMatImplicit method) (nngeometry.object.pspace.PMatKFAC method) (nngeometry.object.pspace.PMatLowRank method) (nngeometry.object.pspace.PMatMixed method) (nngeometry.object.pspace.PMatQuasiDiag method)

get_KFE() (nngeometry.object.pspace.PMatEKFAC method) get_layerid_module_map() (nngeometry.layercollection.LayerCollection method) GroupNormLayer (class in nngeometry.layercollection)

H

has_bias() (nngeometry.layercollection.AbstractLayer method)

L

LayerCollection (class in nngeometry.layercollection)

LayerNormLayer (class in nngeometry.layercollection) LinearLayer (class in nngeometry.layercollection)

M

mapTMmap() (nngeometry.object.pspace.PMatAbstract method) (nngeometry.object.pspace.PMatEKFAC method) (nngeometry.object.pspace.PMatMixed method) merge() (nngeometry.layercollection.LayerCollection method) mm() (nngeometry.object.pspace.PMatBlockDiag method) (nngeometry.object.pspace.PMatDense method) (nngeometry.object.pspace.PMatDiag method) (nngeometry.object.pspace.PMatKFAC method)

mmap() (nngeometry.object.pspace.PMatAbstract method) (nngeometry.object.pspace.PMatImplicit method) module nngeometry.layercollection nngeometry.object.pspace nngeometry.object.vector

N

named_parameters() (nngeometry.layercollection.LayerCollection method) nngeometry.layercollection module nngeometry.object.pspace module nngeometry.object.vector module norm() (nngeometry.object.vector.PVector method) numel() (nngeometry.layercollection.Affine1dLayer method) (nngeometry.layercollection.BatchNorm1dLayer method) (nngeometry.layercollection.BatchNorm2dLayer method) (nngeometry.layercollection.Conv1dLayer method) (nngeometry.layercollection.Conv2dLayer method) (nngeometry.layercollection.ConvTranspose2dLayer method) (nngeometry.layercollection.Cosine1dLayer method) (nngeometry.layercollection.EmbeddingLayer method) (nngeometry.layercollection.GroupNormLayer method) (nngeometry.layercollection.LayerCollection method) (nngeometry.layercollection.LayerNormLayer method) (nngeometry.layercollection.LinearLayer method) (nngeometry.layercollection.Parameter method) (nngeometry.layercollection.RMSNormLayer method) (nngeometry.layercollection.WeightNorm1dLayer method) (nngeometry.layercollection.WeightNorm2dLayer method)

P

Parameter (class in nngeometry.layercollection) parameters() (nngeometry.layercollection.LayerCollection method) PMatAbstract (class in nngeometry.object.pspace) PMatBlockDiag (class in nngeometry.object.pspace) PMatDense (class in nngeometry.object.pspace) PMatDiag (class in nngeometry.object.pspace) PMatEKFAC (class in nngeometry.object.pspace)

PMatEKFACBlockDiag (class in nngeometry.object.pspace) PMatEye (class in nngeometry.object.pspace) PMatImplicit (class in nngeometry.object.pspace) PMatKFAC (class in nngeometry.object.pspace) PMatLowRank (class in nngeometry.object.pspace) PMatMixed (class in nngeometry.object.pspace) PMatQuasiDiag (class in nngeometry.object.pspace) PVector (class in nngeometry.object.vector)

R

random_fvector() (in module nngeometry.object.vector) random_pvector() (in module nngeometry.object.vector)

random_pvector_dict() (in module nngeometry.object.vector) RMSNormLayer (class in nngeometry.layercollection)

S

size() (nngeometry.object.pspace.PMatAbstract method) (nngeometry.object.vector.PVector method) solve() (nngeometry.object.pspace.PMatAbstract method)

solvePFMap() (nngeometry.object.pspace.PMatDense method) solvePVec() (nngeometry.object.pspace.PMatDense method) (nngeometry.object.pspace.PMatDiag method)

T

to_dict() (nngeometry.object.vector.PVector method) to_torch() (nngeometry.object.pspace.PMatEKFAC method) (nngeometry.object.pspace.PMatKFAC method) (nngeometry.object.vector.FVector method) (nngeometry.object.vector.PVector method)

to_torch_layer() (nngeometry.object.vector.PVector method) transposed (nngeometry.layercollection.AbstractLayer attribute) (nngeometry.layercollection.EmbeddingLayer attribute)

U

update_diag() (nngeometry.object.pspace.PMatEKFAC method)

V

vTMv() (nngeometry.object.pspace.PMatAbstract method) (nngeometry.object.pspace.PMatBlockDiag method) (nngeometry.object.pspace.PMatDense method) (nngeometry.object.pspace.PMatDiag method) (nngeometry.object.pspace.PMatEKFAC method) (nngeometry.object.pspace.PMatEye method) (nngeometry.object.pspace.PMatImplicit method) (nngeometry.object.pspace.PMatKFAC method) (nngeometry.object.pspace.PMatLowRank method) (nngeometry.object.pspace.PMatMixed method) (nngeometry.object.pspace.PMatQuasiDiag method)

W

WeightNorm1dLayer (class in nngeometry.layercollection)

WeightNorm2dLayer (class in nngeometry.layercollection)