Index

F

forward() (stadiffuser.models.SpaUNet1DModel method)

G

get_data() (stadiffuser.dataset.MaskNode method)

get_recon() (in module stadiffuser.pipeline) get_slice_loader() (in module stadiffuser.dataset)

M

mask (stadiffuser.dataset.MaskNode attribute) masked_nodes (stadiffuser.dataset.MaskNode attribute) MaskNode (class in stadiffuser.dataset) model (stadiffuser.models.SpaUNet1DModel attribute) module stadiffuser.cli stadiffuser.dataset stadiffuser.models stadiffuser.pipeline

mutual_dict (stadiffuser.dataset.TripletSampler attribute)

N

node_mapping (stadiffuser.dataset.MaskNode attribute)

num_neighbors (stadiffuser.dataset.TripletSampler attribute)

P

prepare_dataset() (in module stadiffuser.pipeline)

pretrain_autoencoder_multi() (in module stadiffuser.pipeline)

Q

query() (stadiffuser.dataset.TripletSampler method)

R

reference_indices (stadiffuser.dataset.TripletSampler attribute) remained_nodes (stadiffuser.dataset.MaskNode attribute)

remove_edge() (in module stadiffuser.pipeline) rng (stadiffuser.dataset.TripletSampler attribute)

S

sample (stadiffuser.models.StaDiffuserOutput attribute) simulate() (in module stadiffuser.pipeline) spatial_coord (stadiffuser.models.StaDiffuserOutput attribute) spatial_embedder (stadiffuser.models.SpaUNet1DModel attribute) SpaUNet1DModel (class in stadiffuser.models) stadiffuser.cli module

stadiffuser.dataset module stadiffuser.models module stadiffuser.pipeline module StaDiffuserOutput (class in stadiffuser.models)

T

target_indices (stadiffuser.dataset.TripletSampler attribute) time_embedder (stadiffuser.models.SpaUNet1DModel attribute) train_autoencoder() (in module stadiffuser.pipeline)

train_autoencoder_multi() (in module stadiffuser.pipeline) train_denoiser() (in module stadiffuser.pipeline) TripletSampler (class in stadiffuser.dataset)