Index

_

__call__() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method)

__version__ (in module pytagi) (in module pytagi.__version)

A

actual_size (pytagi.nn.BaseDeltaStates property) (pytagi.nn.BaseHiddenStates property) (pytagi.nn.data_struct.BaseDeltaStates property) (pytagi.nn.data_struct.BaseHiddenStates property)

AvgPool2d (class in pytagi.nn) (class in pytagi.nn.pooling)

B

backend (pytagi.nn.ddp.DDPConfig property) (pytagi.nn.DDPConfig property) backward() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) barrier() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) BaseDeltaStates (class in pytagi.nn) (class in pytagi.nn.data_struct) BaseHiddenStates (class in pytagi.nn) (class in pytagi.nn.data_struct)

BaseLayer (class in pytagi.nn) (class in pytagi.nn.base_layer) BatchNorm2d (class in pytagi.nn) (class in pytagi.nn.batch_norm) bias (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) block_size (pytagi.nn.BaseDeltaStates property) (pytagi.nn.BaseHiddenStates property) (pytagi.nn.data_struct.BaseDeltaStates property) (pytagi.nn.data_struct.BaseHiddenStates property)

C

classification_error() (in module pytagi.metric) ClosedFormSoftmax (class in pytagi.nn) (class in pytagi.nn.activation) compute_max_min() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method) compute_mean_std() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method)

Conv2d (class in pytagi.nn) (class in pytagi.nn.conv2d) ConvTranspose2d (class in pytagi.nn) (class in pytagi.nn.convtranspose2d) copy_from() (pytagi.nn.BaseDeltaStates method) (pytagi.nn.data_struct.BaseDeltaStates method) create_rolling_window() (pytagi.tagi_utils.Utils method) (pytagi.Utils method)

D

DDPConfig (class in pytagi.nn) (class in pytagi.nn.ddp) DDPSequential (class in pytagi.nn) (class in pytagi.nn.ddp) delta_mu (pytagi.nn.BaseDeltaStates property) (pytagi.nn.data_struct.BaseDeltaStates property) delta_mu_b (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) delta_mu_w (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) delta_var (pytagi.nn.BaseDeltaStates property) (pytagi.nn.data_struct.BaseDeltaStates property) delta_var_b (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property)

delta_var_w (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) delta_z_to_host() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) device (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) (pytagi.nn.output_updater.OutputUpdater property) (pytagi.nn.OutputUpdater property) (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property) device_ids (pytagi.nn.ddp.DDPConfig property) (pytagi.nn.DDPConfig property)

E

Embedding (class in pytagi.nn) (class in pytagi.nn.embedding) error_rate() (pytagi.HRCSoftmaxMetric method) (pytagi.metric.HRCSoftmaxMetric method) eval() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method)

EvenExp (class in pytagi.nn) (class in pytagi.nn.activation) exponential_scheduler() (in module pytagi) (in module pytagi.tagi_utils)

F

forward() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method)

G

get_current_device() (in module pytagi.cuda) get_device_count() (in module pytagi.cuda) get_device_memory() (in module pytagi.cuda) get_device_properties() (in module pytagi.cuda) get_device_with_index() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) get_errors() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) get_hierarchical_softmax() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) get_input_states() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_labels() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) get_layer_info() (pytagi.nn.activation.ClosedFormSoftmax method) (pytagi.nn.activation.EvenExp method) (pytagi.nn.activation.LeakyReLU method) (pytagi.nn.activation.MixtureReLU method) (pytagi.nn.activation.MixtureSigmoid method) (pytagi.nn.activation.MixtureTanh method) (pytagi.nn.activation.ReLU method) (pytagi.nn.activation.Remax method) (pytagi.nn.activation.Sigmoid method) (pytagi.nn.activation.Softmax method) (pytagi.nn.activation.Softplus method) (pytagi.nn.activation.Tanh method) (pytagi.nn.AvgPool2d method) (pytagi.nn.base_layer.BaseLayer method) (pytagi.nn.BaseLayer method) (pytagi.nn.batch_norm.BatchNorm2d method) (pytagi.nn.BatchNorm2d method) (pytagi.nn.ClosedFormSoftmax method) (pytagi.nn.Conv2d method) (pytagi.nn.conv2d.Conv2d method) (pytagi.nn.ConvTranspose2d method) (pytagi.nn.convtranspose2d.ConvTranspose2d method) (pytagi.nn.Embedding method) (pytagi.nn.embedding.Embedding method) (pytagi.nn.EvenExp method) (pytagi.nn.layer_norm.LayerNorm method) (pytagi.nn.LayerNorm method) (pytagi.nn.LeakyReLU method) (pytagi.nn.Linear method) (pytagi.nn.linear.Linear method) (pytagi.nn.LSTM method) (pytagi.nn.lstm.LSTM method) (pytagi.nn.MaxPool2d method) (pytagi.nn.MixtureReLU method) (pytagi.nn.MixtureSigmoid method) (pytagi.nn.MixtureTanh method) (pytagi.nn.pooling.AvgPool2d method) (pytagi.nn.pooling.MaxPool2d method) (pytagi.nn.ReLU method) (pytagi.nn.Remax method) (pytagi.nn.Sigmoid method) (pytagi.nn.SLinear method) (pytagi.nn.slinear.SLinear method) (pytagi.nn.SLSTM method) (pytagi.nn.slstm.SLSTM method) (pytagi.nn.Softmax method) (pytagi.nn.Softplus method) (pytagi.nn.Tanh method) get_layer_name() (pytagi.nn.activation.ClosedFormSoftmax method) (pytagi.nn.activation.EvenExp method) (pytagi.nn.activation.LeakyReLU method) (pytagi.nn.activation.MixtureReLU method) (pytagi.nn.activation.MixtureSigmoid method) (pytagi.nn.activation.MixtureTanh method) (pytagi.nn.activation.ReLU method) (pytagi.nn.activation.Remax method) (pytagi.nn.activation.Sigmoid method) (pytagi.nn.activation.Softmax method) (pytagi.nn.activation.Softplus method) (pytagi.nn.activation.Tanh method) (pytagi.nn.AvgPool2d method) (pytagi.nn.base_layer.BaseLayer method) (pytagi.nn.BaseLayer method) (pytagi.nn.batch_norm.BatchNorm2d method) (pytagi.nn.BatchNorm2d method) (pytagi.nn.ClosedFormSoftmax method) (pytagi.nn.Conv2d method) (pytagi.nn.conv2d.Conv2d method) (pytagi.nn.ConvTranspose2d method) (pytagi.nn.convtranspose2d.ConvTranspose2d method) (pytagi.nn.Embedding method) (pytagi.nn.embedding.Embedding method) (pytagi.nn.EvenExp method) (pytagi.nn.layer_norm.LayerNorm method) (pytagi.nn.LayerNorm method) (pytagi.nn.LeakyReLU method) (pytagi.nn.Linear method) (pytagi.nn.linear.Linear method) (pytagi.nn.LSTM method) (pytagi.nn.lstm.LSTM method) (pytagi.nn.MaxPool2d method) (pytagi.nn.MixtureReLU method) (pytagi.nn.MixtureSigmoid method) (pytagi.nn.MixtureTanh method) (pytagi.nn.pooling.AvgPool2d method) (pytagi.nn.pooling.MaxPool2d method) (pytagi.nn.ReLU method) (pytagi.nn.Remax method) (pytagi.nn.Sigmoid method) (pytagi.nn.SLinear method) (pytagi.nn.slinear.SLinear method) (pytagi.nn.SLSTM method) (pytagi.nn.slstm.SLSTM method) (pytagi.nn.Softmax method) (pytagi.nn.Softplus method) (pytagi.nn.Tanh method)

get_layer_stack_info() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_lstm_states() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_max_num_states() (pytagi.nn.base_layer.BaseLayer method) (pytagi.nn.BaseLayer method) get_name() (pytagi.nn.BaseDeltaStates method) (pytagi.nn.BaseHiddenStates method) (pytagi.nn.data_struct.BaseDeltaStates method) (pytagi.nn.data_struct.BaseHiddenStates method) get_neg_var_w_counter() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_norm_mean_var() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_outputs() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_outputs_smoother() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) get_predicted_labels() (pytagi.HRCSoftmaxMetric method) (pytagi.metric.HRCSoftmaxMetric method) get_upper_triu_cov() (pytagi.tagi_utils.Utils method) (pytagi.Utils method)

H

HRCSoftmax (class in pytagi) (class in pytagi.nn) (class in pytagi.nn.data_struct)

HRCSoftmaxMetric (class in pytagi) (class in pytagi.metric)

I

idx (pytagi.HRCSoftmax property) (pytagi.nn.data_struct.HRCSoftmax property) (pytagi.nn.HRCSoftmax property) in_channels (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) in_height (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) in_width (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) init_input_buffer() (pytagi.nn.resnet_block.ResNetBlock method) (pytagi.nn.ResNetBlock method) init_shortcut_delta_state() (pytagi.nn.resnet_block.ResNetBlock method) (pytagi.nn.ResNetBlock method) init_shortcut_state() (pytagi.nn.resnet_block.ResNetBlock method) (pytagi.nn.ResNetBlock method) init_weight_bias() (pytagi.nn.batch_norm.BatchNorm2d method) (pytagi.nn.BatchNorm2d method) (pytagi.nn.Conv2d method) (pytagi.nn.conv2d.Conv2d method) (pytagi.nn.ConvTranspose2d method) (pytagi.nn.convtranspose2d.ConvTranspose2d method) (pytagi.nn.Embedding method) (pytagi.nn.embedding.Embedding method) (pytagi.nn.layer_norm.LayerNorm method) (pytagi.nn.LayerNorm method) (pytagi.nn.Linear method) (pytagi.nn.linear.Linear method) (pytagi.nn.LSTM method) (pytagi.nn.lstm.LSTM method) (pytagi.nn.SLinear method) (pytagi.nn.slinear.SLinear method) (pytagi.nn.SLSTM method) (pytagi.nn.slstm.SLSTM method)

input_delta_z (pytagi.nn.resnet_block.ResNetBlock property) (pytagi.nn.ResNetBlock property) input_delta_z_buffer (pytagi.nn.ddp.DDPSequential property) (pytagi.nn.DDPSequential property) (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property) input_size (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) input_state_update (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property) input_z (pytagi.nn.resnet_block.ResNetBlock property) (pytagi.nn.ResNetBlock property) is_available() (in module pytagi.cuda) is_device_available() (in module pytagi.cuda) is_nccl_available() (in module pytagi.cuda)

J

jcb (pytagi.nn.BaseHiddenStates property) (pytagi.nn.data_struct.BaseHiddenStates property)

L

label_to_obs() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) label_to_one_hot() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) LayerBlock (class in pytagi.nn) (class in pytagi.nn.layer_block) LayerNorm (class in pytagi.nn) (class in pytagi.nn.layer_norm) layers (pytagi.nn.layer_block.LayerBlock property) (pytagi.nn.LayerBlock property) (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property) LeakyReLU (class in pytagi.nn) (class in pytagi.nn.activation) len (pytagi.HRCSoftmax property) (pytagi.nn.data_struct.HRCSoftmax property) (pytagi.nn.HRCSoftmax property)

Linear (class in pytagi.nn) (class in pytagi.nn.linear) load() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) load_cifar_images() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) load_csv() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) load_mnist_images() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) load_state_dict() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) log_likelihood() (in module pytagi.metric) LSTM (class in pytagi.nn) (class in pytagi.nn.lstm)

M

main_block (pytagi.nn.resnet_block.ResNetBlock property) (pytagi.nn.ResNetBlock property) max_min_norm() (pytagi.Normalizer method) (pytagi.tagi_utils.Normalizer method) max_min_unnorm() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method) max_min_unnorm_std() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method) MaxPool2d (class in pytagi.nn) (class in pytagi.nn.pooling) MixtureReLU (class in pytagi.nn) (class in pytagi.nn.activation) MixtureSigmoid (class in pytagi.nn) (class in pytagi.nn.activation) MixtureTanh (class in pytagi.nn) (class in pytagi.nn.activation) module pytagi pytagi.__version pytagi.cuda pytagi.metric pytagi.nn pytagi.nn.activation pytagi.nn.base_layer pytagi.nn.batch_norm pytagi.nn.conv2d pytagi.nn.convtranspose2d pytagi.nn.data_struct pytagi.nn.ddp pytagi.nn.embedding pytagi.nn.layer_block pytagi.nn.layer_norm pytagi.nn.linear pytagi.nn.lstm pytagi.nn.output_updater pytagi.nn.pooling pytagi.nn.resnet_block pytagi.nn.sequential pytagi.nn.slinear pytagi.nn.slstm pytagi.tagi_utils

mse() (in module pytagi.metric) mu_a (pytagi.nn.BaseHiddenStates property) (pytagi.nn.data_struct.BaseHiddenStates property) mu_b (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) mu_w (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property)

N

Normalizer (class in pytagi) (class in pytagi.tagi_utils) num_biases (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) num_obs (pytagi.HRCSoftmax property) (pytagi.nn.data_struct.HRCSoftmax property) (pytagi.nn.HRCSoftmax property)

num_samples (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property) num_threads (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) num_weights (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property)

O

obs (pytagi.HRCSoftmax property) (pytagi.nn.data_struct.HRCSoftmax property) (pytagi.nn.HRCSoftmax property) obs_to_label_prob() (pytagi.tagi_utils.Utils method) (pytagi.Utils method) out_channels (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) out_height (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) out_width (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) output_delta_z_buffer (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property)

output_size (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) output_to_host() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) output_z_buffer (pytagi.nn.ddp.DDPSequential property) (pytagi.nn.DDPSequential property) (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property) OutputUpdater (class in pytagi.nn) (class in pytagi.nn.output_updater)

P

parameters() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) params_from() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) params_to_device() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) params_to_host() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) preinit_layer() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) pytagi module pytagi.__version module pytagi.cuda module pytagi.metric module pytagi.nn module pytagi.nn.activation module pytagi.nn.base_layer module pytagi.nn.batch_norm module pytagi.nn.conv2d module pytagi.nn.convtranspose2d module

pytagi.nn.data_struct module pytagi.nn.ddp module pytagi.nn.embedding module pytagi.nn.layer_block module pytagi.nn.layer_norm module pytagi.nn.linear module pytagi.nn.lstm module pytagi.nn.output_updater module pytagi.nn.pooling module pytagi.nn.resnet_block module pytagi.nn.sequential module pytagi.nn.slinear module pytagi.nn.slstm module pytagi.tagi_utils module

R

rank (pytagi.nn.ddp.DDPConfig property) (pytagi.nn.DDPConfig property) ReLU (class in pytagi.nn) (class in pytagi.nn.activation) Remax (class in pytagi.nn) (class in pytagi.nn.activation) reset_lstm_states() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method)

reset_zeros() (pytagi.nn.BaseDeltaStates method) (pytagi.nn.data_struct.BaseDeltaStates method) ResNetBlock (class in pytagi.nn) (class in pytagi.nn.resnet_block) rmse() (in module pytagi.metric) ROOT_DIR (in module pytagi.__version)

S

save() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) save_csv() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) Sequential (class in pytagi.nn) (class in pytagi.nn.sequential) set_delta_z() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) set_device() (in module pytagi.cuda) set_input_x() (pytagi.nn.BaseHiddenStates method) (pytagi.nn.data_struct.BaseHiddenStates method) set_lstm_states() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) set_size() (pytagi.nn.BaseDeltaStates method) (pytagi.nn.BaseHiddenStates method) (pytagi.nn.data_struct.BaseDeltaStates method) (pytagi.nn.data_struct.BaseHiddenStates method) set_threads() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) shortcut (pytagi.nn.resnet_block.ResNetBlock property) (pytagi.nn.ResNetBlock property) shortcut_output_delta_z (pytagi.nn.resnet_block.ResNetBlock property) (pytagi.nn.ResNetBlock property) shortcut_output_z (pytagi.nn.resnet_block.ResNetBlock property) (pytagi.nn.ResNetBlock property)

Sigmoid (class in pytagi.nn) (class in pytagi.nn.activation) size (pytagi.nn.BaseDeltaStates property) (pytagi.nn.BaseHiddenStates property) (pytagi.nn.data_struct.BaseDeltaStates property) (pytagi.nn.data_struct.BaseHiddenStates property) SLinear (class in pytagi.nn) (class in pytagi.nn.slinear) SLSTM (class in pytagi.nn) (class in pytagi.nn.slstm) smoother() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) Softmax (class in pytagi.nn) (class in pytagi.nn.activation) Softplus (class in pytagi.nn) (class in pytagi.nn.activation) standardize() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method) state_dict() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) step() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) switch_to_cuda() (pytagi.nn.layer_block.LayerBlock method) (pytagi.nn.LayerBlock method)

T

Tanh (class in pytagi.nn) (class in pytagi.nn.activation) to_cuda() (pytagi.nn.base_layer.BaseLayer method) (pytagi.nn.BaseLayer method) to_device() (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method)

train() (pytagi.nn.ddp.DDPSequential method) (pytagi.nn.DDPSequential method) (pytagi.nn.Sequential method) (pytagi.nn.sequential.Sequential method) training (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property)

U

unstandardize() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method) unstandardize_std() (pytagi.Normalizer static method) (pytagi.tagi_utils.Normalizer static method) update() (pytagi.nn.output_updater.OutputUpdater method) (pytagi.nn.OutputUpdater method)

update_heteros() (pytagi.nn.output_updater.OutputUpdater method) (pytagi.nn.OutputUpdater method) update_using_indices() (pytagi.nn.output_updater.OutputUpdater method) (pytagi.nn.OutputUpdater method) Utils (class in pytagi) (class in pytagi.tagi_utils)

V

var_a (pytagi.nn.BaseHiddenStates property) (pytagi.nn.data_struct.BaseHiddenStates property) var_b (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property)

var_w (pytagi.nn.base_layer.BaseLayer property) (pytagi.nn.BaseLayer property) VERSION (in module pytagi.__version) version (in module pytagi.__version)

W

world_size (pytagi.nn.ddp.DDPConfig property) (pytagi.nn.DDPConfig property)

Z

z_buffer_block_size (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property)

z_buffer_size (pytagi.nn.Sequential property) (pytagi.nn.sequential.Sequential property)