索引

_

__call__() (haiku.AvgPool 方法) (haiku.BatchApply 方法) (haiku.BatchNorm 方法) (haiku.Bias 方法) (haiku.ConvND 方法) (haiku.ConvNDTranspose 方法) (haiku.Deferred 方法) (haiku.EMAParamsTree 方法) (haiku.Embed 方法) (haiku.ExponentialMovingAverage 方法) (haiku.GroupNorm 方法) (haiku.GRU 方法) (haiku.IdentityCore 方法) (haiku.initializers.Constant 方法) (haiku.initializers.Identity 方法) (haiku.initializers.Orthogonal 方法) (haiku.initializers.RandomNormal 方法) (haiku.initializers.RandomUniform 方法) (haiku.initializers.TruncatedNormal 方法) (haiku.initializers.UniformScaling 方法) (haiku.initializers.VarianceScaling 方法) (haiku.LayerNorm 方法) (haiku.Linear 方法) (haiku.LSTM 方法) (haiku.MaxPool 方法) (haiku.MultiHeadAttention 方法) (haiku.nets.MLP 方法) (haiku.nets.MobileNetV1 方法) (haiku.nets.ResNet 方法) (haiku.nets.ResNet.BlockGroup 方法) (haiku.nets.ResNet.BlockV1 方法) (haiku.nets.ResNet.BlockV2 方法) (haiku.nets.VectorQuantizer 方法) (haiku.nets.VectorQuantizerEMA 方法) (haiku.ResetCore 方法) (haiku.Reshape 方法) (haiku.RMSNorm 方法) (haiku.RNNCore 方法) (haiku.SeparableDepthwiseConv2D 方法) (haiku.Sequential 方法) (haiku.SNParamsTree 方法) (haiku.SpectralNorm 方法) (haiku.VanillaRNN 方法) __delattr__() (haiku.Deferred 方法) (haiku.experimental.ArraySpec 方法) (haiku.experimental.MethodInvocation 方法) (haiku.experimental.ModuleDetails 方法) __eq__() (haiku.experimental.ArraySpec 方法) (haiku.experimental.MethodInvocation 方法) (haiku.experimental.ModuleDetails 方法) __hash__() (haiku.experimental.ArraySpec 方法) (haiku.experimental.MethodInvocation 方法) (haiku.experimental.ModuleDetails 方法) __init__() (haiku.AvgPool 方法) (haiku.BatchApply 方法) (haiku.BatchNorm 方法) (haiku.Bias 方法) (haiku.Conv1D 方法) (haiku.Conv1DLSTM 方法) (haiku.Conv1DTranspose 方法) (haiku.Conv2D 方法) (haiku.Conv2DLSTM 方法) (haiku.Conv2DTranspose 方法) (haiku.Conv3D 方法) (haiku.Conv3DLSTM 方法) (haiku.Conv3DTranspose 方法) (haiku.ConvND 方法) (haiku.ConvNDTranspose 方法) (haiku.DeepRNN 方法) (haiku.Deferred 方法) (haiku.DepthwiseConv1D 方法) (haiku.DepthwiseConv2D 方法) (haiku.DepthwiseConv3D 方法) (haiku.EMAParamsTree 方法) (haiku.Embed 方法) (haiku.experimental.ArraySpec 方法) (haiku.experimental.MethodInvocation 方法) (haiku.experimental.ModuleDetails 方法) (haiku.ExponentialMovingAverage 方法) (haiku.Flatten 方法) (haiku.GroupNorm 方法) (haiku.GRU 方法) (haiku.initializers.Constant 方法) (haiku.initializers.Identity 方法) (haiku.initializers.Orthogonal 方法) (haiku.initializers.RandomNormal 方法) (haiku.initializers.RandomUniform 方法) (haiku.initializers.TruncatedNormal 方法) (haiku.initializers.UniformScaling 方法) (haiku.initializers.VarianceScaling 方法) (haiku.InstanceNorm 方法) (haiku.LayerNorm 方法) (haiku.Linear 方法) (haiku.LSTM 方法) (haiku.MaxPool 方法) (haiku.Module 方法) (haiku.MultiHeadAttention 方法) (haiku.nets.MLP 方法) (haiku.nets.MobileNetV1 方法) (haiku.nets.ResNet 方法) (haiku.nets.ResNet.BlockGroup 方法) (haiku.nets.ResNet.BlockV1 方法) (haiku.nets.ResNet.BlockV2 方法) (haiku.nets.ResNet101 方法) (haiku.nets.ResNet152 方法) (haiku.nets.ResNet18 方法) (haiku.nets.ResNet200 方法) (haiku.nets.ResNet34 方法) (haiku.nets.ResNet50 方法) (haiku.nets.VectorQuantizer 方法) (haiku.nets.VectorQuantizerEMA 方法) (haiku.PRNGSequence 方法) (haiku.ResetCore 方法) (haiku.Reshape 方法) (haiku.RMSNorm 方法) (haiku.SeparableDepthwiseConv2D 方法) (haiku.Sequential 方法) (haiku.SNParamsTree 方法) (haiku.SpectralNorm 方法) (haiku.VanillaRNN 方法)

__next__() (haiku.PRNGSequence 方法) __post_init__() (haiku.Module 方法) __setattr__() (haiku.Deferred 方法) (haiku.experimental.ArraySpec 方法) (haiku.experimental.MethodInvocation 方法) (haiku.experimental.ModuleDetails 方法)

A

abstract_to_dot() (在模块 haiku.experimental 中) apply (haiku.MultiTransformed 属性) (haiku.MultiTransformedWithState 属性) (haiku.Transformed 属性) (haiku.TransformedWithState 属性) args_spec (haiku.experimental.MethodInvocation 属性)

ARRAY_INDEX (haiku.EmbedLookupStyle 属性) ArraySpec (haiku.experimental 中的类) as_html() (在模块 haiku.experimental.jaxpr_info 中) as_html_page() (在模块 haiku.experimental.jaxpr_info 中) avg_pool() (在模块 haiku 中) AvgPool (haiku 中的类)

B

BatchApply (haiku 中的类)

BatchNorm (haiku 中的类) Bias (haiku 中的类)

C

call_stack (haiku.experimental.MethodInvocation 属性) causal() (在模块 haiku.pad 中) cell (haiku.LSTMState 属性) check_jax_usage() (在模块 haiku.experimental 中) clear_policy() (在模块 haiku.mixed_precision 中) commitment_cost (haiku.nets.VectorQuantizer 属性) (haiku.nets.VectorQuantizerEMA 属性) cond() (在模块 haiku 中) Constant (haiku.initializers 中的类) context (haiku.experimental.MethodInvocation 属性) context() (在模块 haiku.config 中) Conv1D (haiku 中的类) Conv1DLSTM (haiku 中的类) Conv1DTranspose (haiku 中的类) Conv2D (haiku 中的类)

Conv2DLSTM (haiku 中的类) Conv2DTranspose (haiku 中的类) Conv3D (haiku 中的类) Conv3DLSTM (haiku 中的类) Conv3DTranspose (haiku 中的类) ConvND (haiku 中的类) ConvNDTranspose (haiku 中的类) create() (在模块 haiku.pad 中) create_from_padfn() (在模块 haiku.pad 中) create_from_tuple() (在模块 haiku.pad 中) css() (在模块 haiku.experimental.jaxpr_info 中) current_name() (在模块 haiku 中) current_policy() (在模块 haiku.mixed_precision 中) custom_creator() (在模块 haiku 中) custom_getter() (在模块 haiku 中) custom_setter() (在模块 haiku 中)

D

decay (haiku.nets.VectorQuantizerEMA 属性) deep_rnn_with_skip_connections() (在模块 haiku 中) DeepRNN (haiku 中的类) Deferred (haiku 中的类) DepthwiseConv1D (haiku 中的类)

DepthwiseConv2D (haiku 中的类) DepthwiseConv3D (haiku 中的类) DO_NOT_STORE (在模块 haiku 中) dropout() (在模块 haiku 中) dtype (haiku.experimental.ArraySpec 属性) dynamic_unroll() (在模块 haiku 中)

E

EMAParamsTree (haiku 中的类) Embed (haiku 中的类) embedding_dim (haiku.nets.VectorQuantizer 属性) (haiku.nets.VectorQuantizerEMA 属性) EmbedLookupStyle (haiku 中的类)

epsilon (haiku.nets.VectorQuantizerEMA 属性) eval_shape() (在模块 haiku 中) eval_summary() (在模块 haiku.experimental 中) expand_apply() (在模块 haiku 中) ExponentialMovingAverage (haiku 中的类) Expression (haiku.experimental.jaxpr_info 中的类)

F

fast_eval_shape() (在模块 haiku.experimental 中) filter() (在模块 haiku.data_structures 中) Flatten (haiku 中的类) flatten_flax_to_haiku() (在模块 haiku.experimental.flax 中) force_name() (在模块 haiku 中)

fori_loop() (在模块 haiku 中) format_module() (在模块 haiku.experimental.jaxpr_info 中) full() (在模块 haiku.pad 中) full_name (haiku.GetterContext 属性) (haiku.SetterContext 属性)

G

get_channel_index() (在模块 haiku 中) get_current_state() (在模块 haiku 中) get_initial_state() (在模块 haiku 中) get_parameter() (在模块 haiku 中) get_params() (在模块 haiku 中)

get_policy() (在模块 haiku.mixed_precision 中) get_state() (在模块 haiku 中) GetterContext (haiku 中的类) grad() (在模块 haiku 中) GroupNorm (haiku 中的类) GRU (haiku 中的类)

H

haiku module, [1], [2] haiku.config module haiku.data_structures module haiku.experimental module haiku.experimental.flax module haiku.experimental.jaxpr_info module

haiku.initializers module haiku.mixed_precision module haiku.nets module haiku.pad module haiku.testing module hidden (haiku.LSTMState 属性)

I

Identity (haiku.initializers 中的类) IdentityCore (haiku 中的类) init (haiku.MultiTransformed 属性) (haiku.MultiTransformedWithState 属性) (haiku.Transformed 属性) (haiku.TransformedWithState 属性) initial_state() (haiku.GRU 方法) (haiku.IdentityCore 方法) (haiku.LSTM 方法) (haiku.ResetCore 方法) (haiku.RNNCore 方法) (haiku.VanillaRNN 方法)

initialize() (haiku.ExponentialMovingAverage 方法) Initializer (在模块 haiku.initializers 中) InstanceNorm (haiku 中的类) intercept_methods() (在模块 haiku 中) is_padfn() (在模块 haiku.pad 中) is_subset() (在模块 haiku.data_structures 中)

J

js() (在模块 haiku.experimental.jaxpr_info 中)

K

kwargs_spec (haiku.experimental.MethodInvocation 属性)

L

layer_stack (haiku 中的类) LayerNorm (haiku 中的类) LayerStackTransparencyMapping (haiku 中的类) lift() (在模块 haiku 中) (在模块 haiku.experimental.flax 中) lift_with_state() (在模块 haiku 中)

lifted_prefix_name (haiku.GetterContext 属性) (haiku.SetterContext 属性) LiftWithStateUpdater (haiku 中的类) Linear (haiku 中的类) LSTM (haiku 中的类) LSTMState (haiku 中的类)

M

make_model_info() (在模块 haiku.experimental.jaxpr_info 中) map() (在模块 haiku 中) (在模块 haiku.data_structures 中) max_pool() (在模块 haiku 中) MaxPool (haiku 中的类) maybe_get_rng_sequence_state() (在模块 haiku 中) maybe_next_rng_key() (在模块 haiku 中) merge() (在模块 haiku.data_structures 中) method_name (haiku.experimental.ModuleDetails 属性) (haiku.MethodContext 属性) MethodContext (haiku 中的类) MethodInvocation (haiku.experimental 中的类) MLP (haiku.nets 中的类) MobileNetV1 (haiku.nets 中的类) module haiku, [1], [2] haiku.config haiku.data_structures haiku.experimental haiku.experimental.flax haiku.experimental.jaxpr_info haiku.initializers haiku.mixed_precision haiku.nets haiku.pad haiku.testing

Module (haiku 中的类) (在 haiku.experimental.flax 中的类) (在 haiku.experimental.jaxpr_info 中的类) module (haiku.experimental.ModuleDetails 属性) (haiku.GetterContext 属性) (haiku.MethodContext 属性) (haiku.SetterContext 属性) module_auto_repr() (在模块 haiku.experimental 中) module_details (haiku.experimental.MethodInvocation 属性) module_name (haiku.GetterContext 属性) (haiku.SetterContext 属性) ModuleDetails (haiku.experimental 中的类) ModuleProtocol (haiku 中的类) multi_transform() (在模块 haiku 中) multi_transform_with_state() (在模块 haiku 中) MultiHeadAttention (haiku 中的类) multinomial() (在模块 haiku 中) MultiTransformed (haiku 中的类) MultiTransformedWithState (haiku 中的类) MutableParams (在模块 haiku 中) MutableState (在模块 haiku 中)

N

name (haiku.GetterContext 属性) (haiku.SetterContext 属性) name_like() (在模块 haiku 中) name_scope() (在模块 haiku 中)

next() (haiku.PRNGSequence 方法) next_rng_key() (在模块 haiku 中) next_rng_keys() (在模块 haiku 中) num_embeddings (haiku.nets.VectorQuantizer 属性) (haiku.nets.VectorQuantizerEMA 属性)

O

ONE_HOT (haiku.EmbedLookupStyle 属性) one_hot() (在模块 haiku 中) optimize_rng_use() (在模块 haiku.experimental 中) orig_class (haiku.MethodContext 属性) orig_method (haiku.MethodContext 属性) original_dtype (haiku.GetterContext 属性) (haiku.SetterContext 属性)

original_init (haiku.GetterContext 属性) original_shape (haiku.GetterContext 属性) (haiku.SetterContext 属性) Orthogonal (haiku.initializers 中的类) output_spec (haiku.experimental.MethodInvocation 属性)

P

PadFn (在模块 haiku.pad 中) params (haiku.experimental.ModuleDetails 属性) Params (在模块 haiku 中) params_dict() (haiku.Module 方法)

partition() (在模块 haiku.data_structures 中) partition_n() (在模块 haiku.data_structures 中) PRNGSequence (haiku 中的类) push_policy() (在模块 haiku.mixed_precision 中)

Q

quantize() (haiku.nets.VectorQuantizer 方法) (haiku.nets.VectorQuantizerEMA 方法)

R

RandomNormal (haiku.initializers 中的类) RandomUniform (haiku.initializers 中的类) remat() (在模块 haiku 中) replace_rng_sequence_state() (在模块 haiku 中) reserve() (haiku.PRNGSequence 方法) reserve_rng_keys() (在模块 haiku 中) ResetCore (haiku 中的类) Reshape (haiku 中的类) ResNet (haiku.nets 中的类) ResNet.BlockGroup (haiku.nets 中的类) ResNet.BlockV1 (haiku.nets 中的类) ResNet.BlockV2 (haiku.nets 中的类)

ResNet101 (haiku.nets 中的类) ResNet152 (haiku.nets 中的类) ResNet18 (haiku.nets 中的类) ResNet200 (haiku.nets 中的类) ResNet34 (haiku.nets 中的类) ResNet50 (haiku.nets 中的类) reverse() (haiku.nets.MLP 方法) reverse_causal() (在模块 haiku.pad 中) RMSNorm (haiku 中的类) rng_reserve_size() (在模块 haiku.experimental 中) RNNCore (haiku 中的类) running_init() (在模块 haiku 中)

S

same() (在模块 haiku.pad 中) scan() (在模块 haiku 中) SeparableDepthwiseConv2D (haiku 中的类) Sequential (haiku 中的类) set() (在模块 haiku.config 中) set_policy() (在模块 haiku.mixed_precision 中) set_state() (在模块 haiku 中) SetterContext (haiku 中的类)

shape (haiku.experimental.ArraySpec 属性) SNParamsTree (haiku 中的类) SpectralNorm (haiku 中的类) state (haiku.experimental.ModuleDetails 属性) State (在模块 haiku 中) state_dict() (haiku.Module 方法) static_unroll() (在模块 haiku 中) SupportsCall (haiku 中的类) switch() (在模块 haiku 中)

T

tabulate() (在模块 haiku.experimental 中) target (haiku.Deferred 属性) to_dot() (在模块 haiku 中) to_haiku_dict() (在模块 haiku.data_structures 中) to_immutable_dict() (在模块 haiku.data_structures 中) to_module() (在模块 haiku 中) to_mutable_dict() (在模块 haiku.data_structures 中) transform() (在模块 haiku 中) transform_and_run() (在模块 haiku.testing 中)

transform_with_state() (在模块 haiku 中) Transformed (haiku 中的类) TransformedWithState (haiku 中的类) transparent() (在模块 haiku 中) transparent_lift() (在模块 haiku 中) transparent_lift_with_state() (在模块 haiku 中) traverse() (在模块 haiku.data_structures 中) tree_bytes() (在模块 haiku.data_structures 中) tree_size() (在模块 haiku.data_structures 中) TruncatedNormal (haiku.initializers 中的类)

U

UniformScaling (haiku.initializers 中的类)

V

valid() (在模块 haiku.pad 中) value_and_grad() (在模块 haiku 中) VanillaRNN (haiku 中的类)

VarianceScaling (haiku.initializers 中的类) VectorQuantizer (haiku.nets 中的类) VectorQuantizerEMA (haiku.nets 中的类) vmap() (在模块 haiku 中)

W

while_loop() (在模块 haiku 中) with_empty_state() (在模块 haiku 中)

with_rng() (在模块 haiku 中) without_apply_rng() (在模块 haiku 中) without_state() (在模块 haiku 中)