SYMBOL INDEX (1832 symbols across 170 files) FILE: conftest.py function pytest_sessionstart (line 25) | def pytest_sessionstart(session): function pytest_sessionfinish (line 30) | def pytest_sessionfinish(session, exitstatus): function add_modules (line 37) | def add_modules(doctest_namespace): FILE: docs/source/conf.py function class_inherit_diagrams (line 52) | def class_inherit_diagrams(_): function process_docstring (line 62) | def process_docstring(app, what, name, obj, options, lines): function setup (line 80) | def setup(app): FILE: docs/source/conf/25-napoleon.py function parse_keys_section (line 9) | def parse_keys_section(self, section): function parse_attributes_section (line 16) | def parse_attributes_section(self, section): function parse_class_attributes_section (line 23) | def parse_class_attributes_section(self, section): function patched_parse (line 32) | def patched_parse(self): FILE: docs/source/conf/85-dtype_typehints.py function typehints_formatter_function (line 29) | def typehints_formatter_function(annotation, config): FILE: docs/source/docsutil.py function run_conf_files (line 20) | def run_conf_files(vardict=None, path=None): function package_classes (line 47) | def package_classes(package): function get_text_indentation (line 87) | def get_text_indentation(text, skiplines=0): function add_text_indentation (line 110) | def add_text_indentation(text, indent): function insert_inheritance_diagram (line 127) | def insert_inheritance_diagram(clsqname, parts=None, default_nparts=2): FILE: docs/source/pyfigures/xray_3d_vec.py class Arrow3D (line 12) | class Arrow3D(FancyArrowPatch): method __init__ (line 13) | def __init__(self, xs, ys, zs, *args, **kwargs): method do_3d_projection (line 17) | def do_3d_projection(self, renderer=None): FILE: docs/source/pyfigures/xray_3d_vol.py class Arrow3D (line 12) | class Arrow3D(FancyArrowPatch): method __init__ (line 13) | def __init__(self, xs, ys, zs, *args, **kwargs): method do_3d_projection (line 17) | def do_3d_projection(self, renderer=None): FILE: examples/jnb.py function py_file_to_string (line 18) | def py_file_to_string(src): function script_to_notebook (line 83) | def script_to_notebook(src, dst): function read_notebook (line 91) | def read_notebook(fname): function execute_notebook (line 101) | def execute_notebook(fname): function notebook_executed (line 120) | def notebook_executed(nbfn): function same_notebook_code (line 134) | def same_notebook_code(nb1, nb2): function same_notebook_markdown (line 166) | def same_notebook_markdown(nb1, nb2): function replace_markdown_cells (line 198) | def replace_markdown_cells(src, dst): function remove_error_output (line 229) | def remove_error_output(src): FILE: examples/makenotebooks.py function script_uses_ray (line 30) | def script_uses_ray(fname): function script_path (line 40) | def script_path(sname): function notebook_path (line 46) | def notebook_path(sname): function ray_run_nb (line 190) | def ray_run_nb(fname): FILE: examples/scripts/ct_abel_tv_admm_tune.py class Trainable (line 77) | class Trainable(tune.Trainable): method setup (line 80) | def setup(self, config, x_gt, x0, y): method reset_config (line 96) | def reset_config(self, config): method step (line 127) | def step(self): FILE: examples/scripts/ct_astra_weighted_tv_admm.py function postprocess (line 80) | def postprocess(x): function plot_recon (line 157) | def plot_recon(x, title, ax): FILE: examples/scripts/ct_fan_svmbir_ppp_bm3d_admm_prox.py function add_poisson_noise (line 93) | def add_poisson_noise(sino, max_intensity): FILE: examples/scripts/deconv_microscopy_allchn_tv_admm.py function deconvolve_channel (line 110) | def deconvolve_channel(channel): FILE: examples/scripts/deconv_tv_admm_tune.py function eval_params (line 76) | def eval_params(config, x_gt, psf, y): FILE: examples/scripts/demosaic_ppp_bm3d_admm.py function Afn (line 50) | def Afn(x): function ATfn (line 63) | def ATfn(x): function demosaic (line 83) | def demosaic(cfaimg): FILE: examples/scripts/denoise_tv_apgm.py class DualTVLoss (line 81) | class DualTVLoss(loss.Loss): method __init__ (line 82) | def __init__( method __call__ (line 92) | def __call__(self, x: Union[Array, BlockArray]) -> float: class IsoProjector (line 104) | class IsoProjector(functional.Functional): method __call__ (line 108) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 111) | def prox(self, v: Array, lam: float, **kwargs) -> Array: class AnisoProjector (line 156) | class AnisoProjector(functional.Functional): method __call__ (line 160) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 163) | def prox(self, v: Array, lam: float, **kwargs) -> Array: FILE: examples/scripts/sparsecode_poisson_apgm.py class ForwardOperator (line 71) | class ForwardOperator(Operator): method __init__ (line 80) | def __init__(self, input_shape: Shape, D0, D1, jit: bool = True): method _eval (line 94) | def _eval(self, x: BlockArray) -> BlockArray: function plot_results (line 128) | def plot_results(hist, str_ss, L0, xsol, xgt, Aop): FILE: examples/scripts/superres_ppp_dncnn_admm.py function downsample_image (line 31) | def downsample_image(img, rate): FILE: examples/updatejnbcode.py function replace_code_cells (line 15) | def replace_code_cells(src, dst): FILE: scico/_core.py function _append_jax_docs (line 31) | def _append_jax_docs(fn, jaxfn=None): function _convert_ba_dts (line 46) | def _convert_ba_dts(arg: Any) -> Any: function eval_shape (line 58) | def eval_shape(fun: Callable, *args, **kwargs) -> Any: function grad (line 71) | def grad( function value_and_grad (line 99) | def value_and_grad( function linear_transpose (line 128) | def linear_transpose(fun: Callable, *primals) -> Callable: function linear_adjoint (line 140) | def linear_adjoint(fun: Callable, *primals) -> Callable: function jacrev (line 156) | def jacrev( function cvjp (line 177) | def cvjp(fun: Callable, *primals, jidx: Optional[int] = None) -> Tuple[T... FILE: scico/_version.py function root_init_path (line 17) | def root_init_path() -> str: # pragma: no cover function variable_assign_value (line 26) | def variable_assign_value(path: str, var: str) -> Any: function init_variable_assign_value (line 50) | def init_variable_assign_value(var: str) -> Any: # pragma: no cover function current_git_hash (line 66) | def current_git_hash() -> Optional[str]: # nosec pragma: no cover function package_version (line 79) | def package_version(split: bool = False) -> Union[str, Tuple[str, str]]:... FILE: scico/data/__init__.py function _imread (line 20) | def _imread(filename: str, path: Optional[str] = None, asfloat: bool = F... function kodim23 (line 41) | def kodim23(asfloat: bool = False) -> snp.Array: function _flax_data_path (line 55) | def _flax_data_path(filename: str) -> str: FILE: scico/denoiser.py function bm3d (line 39) | def bm3d(x: snp.Array, sigma: float, is_rgb: bool = False, profile: Unio... function bm4d (line 117) | def bm4d(x: snp.Array, sigma: float, profile: Union[BM4DProfile, str] = ... class DnCNN (line 182) | class DnCNN(FlaxMap): method __init__ (line 202) | def __init__(self, variant: str = "6M"): method __call__ (line 239) | def __call__(self, x: snp.Array, sigma: Optional[float] = None) -> snp... FILE: scico/diagnostics.py class IterationStats (line 18) | class IterationStats: method __init__ (line 25) | def __init__( method insert (line 156) | def insert(self, values: Union[List, Tuple]): method end (line 180) | def end(self): method history (line 196) | def history(self, transpose: bool = False) -> Union[List[NamedTuple], ... FILE: scico/examples.py function rgb2gray (line 30) | def rgb2gray(rgb: np.ndarray) -> np.ndarray: function volume_read (line 49) | def volume_read(path: str, ext: str = "tif") -> np.ndarray: function get_epfl_deconv_data (line 72) | def get_epfl_deconv_data(channel: int, path: str, verbose: bool = False)... function epfl_deconv_data (line 128) | def epfl_deconv_data( function get_ucb_diffusercam_data (line 168) | def get_ucb_diffusercam_data(path: str, verbose: bool = False): # pragm... function ucb_diffusercam_data (line 221) | def ucb_diffusercam_data( function downsample_volume (line 260) | def downsample_volume(vol: np.ndarray, rate: int) -> np.ndarray: function tile_volume_slices (line 289) | def tile_volume_slices(x: np.ndarray, sep_width: int = 10) -> np.ndarray: function gaussian (line 347) | def gaussian(shape: Shape, sigma: Optional[np.ndarray] = None) -> np.nda... function create_cone (line 384) | def create_cone(shape: Shape, center: Optional[List[float]] = None) -> n... function create_circular_phantom (line 409) | def create_circular_phantom( function create_3d_foam_phantom (line 439) | def create_3d_foam_phantom( function create_conv_sparse_phantom (line 488) | def create_conv_sparse_phantom(Nx: int, Nnz: int) -> Tuple[np.ndarray, n... function create_tangle_phantom (line 529) | def create_tangle_phantom(nx: int, ny: int, nz: int) -> np.ndarray: function create_block_phantom (line 567) | def create_block_phantom(out_shape: Shape) -> np.ndarray: function spnoise (line 604) | def spnoise( function phase_diff (line 632) | def phase_diff(x: snp.Array, y: snp.Array) -> snp.Array: FILE: scico/flax/_flax.py function load_variables (line 23) | def load_variables(filename: str) -> PyTree: function save_variables (line 43) | def save_variables(variables: PyTree, filename: str): class FlaxMap (line 57) | class FlaxMap: method __init__ (line 60) | def __init__(self, model: Module, variables: PyTree): method __call__ (line 71) | def __call__(self, x: Array) -> Array: FILE: scico/flax/_models.py class DnCNNNet (line 41) | class DnCNNNet(Module): method __call__ (line 67) | def __call__( class ResNet (line 121) | class ResNet(Module): method __call__ (line 147) | def __call__(self, x: Array, train: bool = True) -> Array: class ConvBNNet (line 196) | class ConvBNNet(Module): method __call__ (line 221) | def __call__(self, x: Array, train: bool = True) -> Array: class UNet (line 267) | class UNet(Module): method __call__ (line 295) | def __call__(self, x: Array, train: bool = True) -> Array: FILE: scico/flax/blocks.py class ConvBNBlock (line 30) | class ConvBNBlock(Module): method __call__ (line 55) | def __call__( class ConvBlock (line 76) | class ConvBlock(Module): method __call__ (line 98) | def __call__( class ConvBNPoolBlock (line 118) | class ConvBNPoolBlock(Module): method __call__ (line 147) | def __call__( class ConvBNUpsampleBlock (line 170) | class ConvBNUpsampleBlock(Module): method __call__ (line 196) | def __call__( class ConvBNMultiBlock (line 218) | class ConvBNMultiBlock(Module): method __call__ (line 247) | def __call__( function upscale_nn (line 275) | def upscale_nn(x: Array, scale: int = 2) -> Array: FILE: scico/flax/examples/data_generation.py class UnitCircle (line 25) | class UnitCircle: class Foam2 (line 61) | class Foam2(UnitCircle): method __init__ (line 67) | def __init__( function generate_foam1_images (line 101) | def generate_foam1_images(seed: float, size: int, ndata: int) -> np.ndar... function generate_foam2_images (line 126) | def generate_foam2_images(seed: float, size: int, ndata: int) -> np.ndar... function vector_f (line 153) | def vector_f(f_: Callable, v: Array) -> Array: function batched_f (line 168) | def batched_f(f_: Callable, vr: Array) -> Array: function generate_ct_data (line 189) | def generate_ct_data( function generate_blur_data (line 276) | def generate_blur_data( function distributed_data_generation (line 353) | def distributed_data_generation( FILE: scico/flax/examples/data_preprocessing.py function rotation90 (line 33) | def rotation90(img: Array) -> Array: function flip (line 54) | def flip(img: Array) -> Array: class CenterCrop (line 74) | class CenterCrop: method __init__ (line 81) | def __init__(self, output_size: Union[Shape, int]): method __call__ (line 94) | def __call__(self, image: Array) -> Array: class PositionalCrop (line 114) | class PositionalCrop: method __init__ (line 121) | def __init__(self, output_size: Union[Shape, int]): method __call__ (line 134) | def __call__(self, image: Array, top: int, left: int) -> Array: class RandomNoise (line 155) | class RandomNoise: method __init__ (line 168) | def __init__(self, noise_level: float, range_flag: bool = False): method __call__ (line 181) | def __call__(self, image: Array) -> Array: function preprocess_images (line 209) | def preprocess_images( function build_image_dataset (line 291) | def build_image_dataset( function images_read (line 371) | def images_read(path: str, ext: str = "jpg") -> Array: # pragma: no cover function get_bsds_data (line 401) | def get_bsds_data(path: str, verbose: bool = False): # pragma: no cover function build_blur_kernel (line 468) | def build_blur_kernel( class PaddedCircularConvolve (line 492) | class PaddedCircularConvolve(LinearOperator): method __init__ (line 501) | def __init__( method _eval (line 560) | def _eval(self, x: Array) -> Array: FILE: scico/flax/examples/examples.py function get_cache_path (line 24) | def get_cache_path(cache_path: Optional[str] = None) -> Tuple[str, str]: function load_ct_data (line 45) | def load_ct_data( function load_blur_data (line 177) | def load_blur_data( function load_image_data (line 328) | def load_image_data( function check_img_data_requirements (line 513) | def check_img_data_requirements( function print_input_path (line 576) | def print_input_path(path_display: str): # pragma: no cover function print_output_path (line 585) | def print_output_path(path_display: str): # pragma: no cover function print_data_range (line 594) | def print_data_range(idstring: str, data: Array): # pragma: no cover function print_data_size (line 604) | def print_data_size(idstring: str, size: int): # pragma: no cover function print_info (line 614) | def print_info( function print_data_warning (line 637) | def print_data_warning(idstring: str, requested: int, available: int): ... function runtime_error_scalar (line 651) | def runtime_error_scalar( function runtime_error_array (line 672) | def runtime_error_array(type: str, idstring: str, maxdiff: float): FILE: scico/flax/examples/typed_dict.py class CTDataSetDict (line 22) | class CTDataSetDict(TypedDict): class ConfigImageSetDict (line 30) | class ConfigImageSetDict(TypedDict): FILE: scico/flax/inverse.py class MoDLNet (line 35) | class MoDLNet(Module): method __call__ (line 68) | def __call__(self, y: Array, train: bool = True) -> Array: function cg_solver (line 113) | def cg_solver(A: Callable, b: Array, x0: Array = None, maxiter: int = 50... class ODPProxDnBlock (line 161) | class ODPProxDnBlock(Module): method batch_op_adj (line 191) | def batch_op_adj(self, y: Array) -> Array: method __call__ (line 196) | def __call__(self, x: Array, y: Array, train: bool = True) -> Array: class ODPProxDcnvBlock (line 228) | class ODPProxDcnvBlock(Module): method setup (line 258) | def setup(self): method batch_op_adj (line 281) | def batch_op_adj(self, y: Array) -> Array: method __call__ (line 285) | def __call__(self, x: Array, y: Array, train: bool = True) -> Array: class ODPGrDescBlock (line 318) | class ODPGrDescBlock(Module): method setup (line 349) | def setup(self): method batch_op_adj (line 370) | def batch_op_adj(self, y: Array) -> Array: method __call__ (line 374) | def __call__(self, x: Array, y: Array, train: bool = True) -> Array: class ODPNet (line 392) | class ODPNet(Module): method __call__ (line 428) | def __call__(self, y: Array, train: bool = True) -> Array: function power_iteration (line 465) | def power_iteration(A: LinearOperator, maxiter: int = 100): FILE: scico/flax/train/apply.py function apply_fn (line 31) | def apply_fn(model: ModuleDef, variables: ModelVarDict, batch: DataSetDi... function only_apply (line 52) | def only_apply( FILE: scico/flax/train/checkpoints.py function checkpoint_restore (line 38) | def checkpoint_restore( function checkpoint_save (line 84) | def checkpoint_save(state: TrainState, config: ConfigDict, workdir: Unio... FILE: scico/flax/train/clu_utils.py class ParamRow (line 36) | class ParamRow: class ParamRowWithStats (line 45) | class ParamRowWithStats(ParamRow): function flatten_dict (line 52) | def flatten_dict( function count_parameters (line 75) | def count_parameters(params: PyTree) -> int: function get_parameter_rows (line 88) | def get_parameter_rows( function _default_table_value_formatter (line 129) | def _default_table_value_formatter(value): function make_table (line 141) | def make_table( function get_parameter_overview (line 209) | def get_parameter_overview( FILE: scico/flax/train/diagnostics.py function compute_metrics (line 25) | def compute_metrics(output: Array, labels: Array, criterion: Callable = ... class ArgumentStruct (line 49) | class ArgumentStruct: method __init__ (line 57) | def __init__(self, **entries): function stats_obj (line 61) | def stats_obj() -> Tuple[IterationStats, Callable]: FILE: scico/flax/train/input_pipeline.py class IterateData (line 32) | class IterateData: method __init__ (line 38) | def __init__(self, dt: DataSetDict, batch_size: int, train: bool = Tru... method reset (line 60) | def reset(self): method __iter__ (line 72) | def __iter__(self): method __next__ (line 75) | def __next__(self): function prepare_data (line 90) | def prepare_data(xs: Array) -> Any: function create_input_iter (line 102) | def create_input_iter( FILE: scico/flax/train/learning_rate.py function create_cnst_lr_schedule (line 15) | def create_cnst_lr_schedule(config: ConfigDict) -> optax._src.base.Sched... function create_exp_lr_schedule (line 30) | def create_exp_lr_schedule(config: ConfigDict) -> optax._src.base.Schedule: function create_cosine_lr_schedule (line 48) | def create_cosine_lr_schedule(config: ConfigDict) -> optax._src.base.Sch... FILE: scico/flax/train/losses.py function mse_loss (line 17) | def mse_loss(output: Array, labels: Array) -> float: FILE: scico/flax/train/spectral.py function _l2_normalize (line 35) | def _l2_normalize(x: Array, eps: float = 1e-12) -> Array: function estimate_spectral_norm (line 49) | def estimate_spectral_norm( class CNN (line 85) | class CNN(Module): method __call__ (line 104) | def __call__(self, x): function conv (line 126) | def conv(inputs: Array, kernel: Array) -> Array: function spectral_normalization_conv (line 151) | def spectral_normalization_conv( function exact_spectral_norm (line 177) | def exact_spectral_norm(f, input_shape): FILE: scico/flax/train/state.py class TrainState (line 29) | class TrainState(train_state.TrainState): function initialize (line 39) | def initialize(key: KeyArray, model: ModuleDef, ishape: Shape) -> Tuple[... function create_basic_train_state (line 63) | def create_basic_train_state( FILE: scico/flax/train/steps.py function train_step (line 26) | def train_step( function train_step_post (line 93) | def train_step_post( function eval_step (line 139) | def eval_step( FILE: scico/flax/train/trainer.py function sync_batch_stats (line 47) | def sync_batch_stats(state: TrainState) -> TrainState: class BasicFlaxTrainer (line 59) | class BasicFlaxTrainer: method __init__ (line 62) | def __init__( method set_training_parameters (line 111) | def set_training_parameters( method configure_steps (line 132) | def configure_steps( method configure_reporting (line 183) | def configure_reporting(self, config: ConfigDict): method configure_training_functions (line 229) | def configure_training_functions(self, config: ConfigDict): method construct_data_iterators (line 296) | def construct_data_iterators( method define_parallel_training_functions (line 344) | def define_parallel_training_functions(self): method initialize_training_state (line 379) | def initialize_training_state( method train (line 414) | def train(self) -> Tuple[Dict[str, Any], Optional[IterationStats]]: method update_metrics (line 476) | def update_metrics(self, state: TrainState, step: int, train_metrics: ... method checkpoint (line 523) | def checkpoint(self, state: TrainState): # pragma: no cover method log (line 532) | def log(self, logstr: str): FILE: scico/flax/train/traversals.py function construct_traversal (line 19) | def construct_traversal(prmname: str) -> ModelParamTraversal: function clip_positive (line 31) | def clip_positive(params: PyTree, traversal: ModelParamTraversal, minval... function clip_range (line 45) | def clip_range( FILE: scico/flax/train/typed_dict.py class DataSetDict (line 23) | class DataSetDict(TypedDict): class ConfigDict (line 36) | class ConfigDict(TypedDict): class ModelVarDict (line 94) | class ModelVarDict(TypedDict): class MetricsDict (line 107) | class MetricsDict(TypedDict, total=False): FILE: scico/function.py class Function (line 23) | class Function: method __init__ (line 32) | def __init__( method __repr__ (line 90) | def __repr__(self): method __call__ (line 98) | def __call__(self, *args: Union[Array, BlockArray]) -> Union[Array, Bl... method slice (line 109) | def slice(self, index: int, *fix_args: Union[Array, BlockArray]) -> Op... method join (line 134) | def join(self) -> Operator: method jvp (line 164) | def jvp( method vjp (line 191) | def vjp( method jacobian (line 217) | def jacobian( FILE: scico/functional/_denoiser.py class BM3D (line 18) | class BM3D(Functional): method __init__ (line 31) | def __init__(self, is_rgb: bool = False, profile: Union[denoiser.BM3DP... method prox (line 44) | def prox(self, x: Array, lam: float = 1.0, **kwargs) -> Array: # type... class BM4D (line 59) | class BM4D(Functional): method __init__ (line 72) | def __init__(self, profile: Union[denoiser.BM4DProfile, str] = "np"): method prox (line 81) | def prox(self, x: Array, lam: float = 1.0, **kwargs) -> Array: # type... class DnCNN (line 96) | class DnCNN(Functional): method __init__ (line 107) | def __init__(self, variant: str = "6M"): method prox (line 126) | def prox(self, x: Array, lam: float = 1.0, **kwargs) -> Array: # type... FILE: scico/functional/_dist.py class SetDistance (line 18) | class SetDistance(Functional): method __init__ (line 41) | def __init__(self, proj: Callable, args=()): method __call__ (line 50) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 65) | def prox( class SquaredSetDistance (line 88) | class SquaredSetDistance(Functional): method __init__ (line 112) | def __init__(self, proj: Callable, args=()): method __call__ (line 121) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 136) | def prox( FILE: scico/functional/_functional.py class Functional (line 22) | class Functional: method __init__ (line 38) | def __init__(self): method __str__ (line 41) | def __str__(self): method __repr__ (line 44) | def __repr__(self): method __mul__ (line 47) | def __mul__(self, other: Union[float, int]) -> ScaledFunctional: method __rmul__ (line 52) | def __rmul__(self, other: Union[float, int]) -> ScaledFunctional: method __add__ (line 55) | def __add__(self, other: Functional) -> FunctionalSum: method __call__ (line 60) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 72) | def prox( method conj_prox (line 102) | def conj_prox( method grad (line 130) | def grad(self, x: Union[Array, BlockArray]): class ScaledFunctional (line 142) | class ScaledFunctional(Functional): method __init__ (line 145) | def __init__(self, functional: Functional, scale: float): method __repr__ (line 152) | def __repr__(self): method __call__ (line 159) | def __call__(self, x: Union[Array, BlockArray]) -> float: method __mul__ (line 162) | def __mul__(self, other: Union[float, int]) -> ScaledFunctional: method prox (line 167) | def prox( class SeparableFunctional (line 201) | class SeparableFunctional(Functional): method __init__ (line 216) | def __init__(self, functional_list: List[Functional]): method __repr__ (line 228) | def __repr__(self): method __call__ (line 236) | def __call__(self, x: BlockArray) -> float: method prox (line 244) | def prox(self, v: BlockArray, lam: float = 1.0, **kwargs) -> BlockArray: class ComposedFunctional (line 277) | class ComposedFunctional(Functional): method __init__ (line 294) | def __init__(self, functional: Functional, linop: LinearOperator): method __repr__ (line 309) | def __repr__(self): method __call__ (line 315) | def __call__(self, x: BlockArray) -> float: method prox (line 318) | def prox( class FunctionalSum (line 345) | class FunctionalSum(Functional): method __init__ (line 348) | def __init__(self, functional1: Functional, functional2: Functional): method __repr__ (line 355) | def __repr__(self): method __call__ (line 362) | def __call__(self, x: Union[Array, BlockArray]) -> float: class ZeroFunctional (line 366) | class ZeroFunctional(Functional): method __call__ (line 372) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 375) | def prox( FILE: scico/functional/_indicator.py class NonNegativeIndicator (line 21) | class NonNegativeIndicator(Functional): method __call__ (line 37) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 44) | def prox( class L2BallIndicator (line 71) | class L2BallIndicator(Functional): method __init__ (line 89) | def __init__(self, radius: float = 1.0): method __call__ (line 98) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 102) | def prox( class BoxIndicator (line 130) | class BoxIndicator(Functional): method __init__ (line 140) | def __init__(self, lb: float = 0.0, ub: float = 1.0): method __call__ (line 150) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 156) | def prox( FILE: scico/functional/_norm.py class L0Norm (line 23) | class L0Norm(Functional): method __call__ (line 35) | def __call__(x: Union[Array, BlockArray]) -> float: method prox (line 40) | def prox(v: Union[Array, BlockArray], lam: float = 1.0, **kwargs) -> U... class L1Norm (line 65) | class L1Norm(Functional): method __call__ (line 79) | def __call__(x: Union[Array, BlockArray]) -> float: method prox (line 84) | def prox(v: Union[Array, BlockArray], lam: float = 1.0, **kwargs) -> A... class SquaredL2Norm (line 119) | class SquaredL2Norm(Functional): method __call__ (line 133) | def __call__(x: Union[Array, BlockArray]) -> float: method prox (line 140) | def prox(v: Union[Array, BlockArray], lam: float = 1.0, **kwargs) -> U... class L2Norm (line 161) | class L2Norm(Functional): method __call__ (line 173) | def __call__(x: Union[Array, BlockArray]) -> float: method prox (line 178) | def prox(v: Union[Array, BlockArray], lam: float = 1.0, **kwargs) -> U... class L21Norm (line 208) | class L21Norm(Functional): method __init__ (line 229) | def __init__(self, l2_axis: Union[None, int, Tuple] = 0): method _l2norm (line 240) | def _l2norm( method __call__ (line 246) | def __call__(self, x: Union[Array, BlockArray]) -> float: method _prox (line 254) | def _prox( method prox (line 265) | def prox( class L1MinusL2Norm (line 299) | class L1MinusL2Norm(Functional): method __init__ (line 311) | def __init__(self, beta: float = 1.0): method _l1minusl2norm (line 320) | def _l1minusl2norm(x: Union[Array, BlockArray], beta: float) -> float: method __call__ (line 324) | def __call__(self, x: Union[Array, BlockArray]) -> float: method _prox_vamx_ge_thresh (line 328) | def _prox_vamx_ge_thresh(v, va, vs, alpha, beta): method _prox_vamx_le_alpha (line 337) | def _prox_vamx_le_alpha(v, va, vs, vamx, alpha, beta): method _prox_vamx_gt_alpha (line 345) | def _prox_vamx_gt_alpha(v, va, vs, alpha, beta): method _prox_vamx_gt_0 (line 352) | def _prox_vamx_gt_0(v, va, vs, vamx, alpha, beta): method _prox (line 361) | def _prox(v: Union[Array, BlockArray], lam: float, beta: float) -> Uni... method prox (line 377) | def prox( class HuberNorm (line 400) | class HuberNorm(Functional): method __init__ (line 435) | def __init__(self, delta: float = 1.0, separable: bool = True): method _call_sep (line 456) | def _call_sep(x: Union[Array, BlockArray], delta: float) -> float: method _call_nonsep (line 463) | def _call_nonsep(x: Union[Array, BlockArray], delta: float) -> float: method __call__ (line 469) | def __call__(self, x: Union[Array, BlockArray]) -> float: method _prox_sep (line 474) | def _prox_sep( method _prox_nonsep (line 482) | def _prox_nonsep( method prox (line 489) | def prox( class NuclearNorm (line 523) | class NuclearNorm(Functional): method __call__ (line 539) | def __call__(x: Union[Array, BlockArray]) -> float: method prox (line 546) | def prox(v: Union[Array, BlockArray], lam: float = 1.0, **kwargs) -> U... FILE: scico/functional/_proxavg.py class ProximalAverage (line 17) | class ProximalAverage(Functional): method __init__ (line 36) | def __init__( method __repr__ (line 76) | def __repr__(self): method __call__ (line 86) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 98) | def prox( FILE: scico/functional/_tvnorm.py class TVNorm (line 33) | class TVNorm(Functional): method __init__ (line 44) | def __init__( method _call_operator (line 89) | def _call_operator(self, input_shape: Shape, input_dtype: DType) -> Li... method __call__ (line 104) | def __call__(self, x: Array) -> float: method _prox_operators (line 117) | def _prox_operators( method _slice_tuple_to_tuple (line 164) | def _slice_tuple_to_tuple(st: Tuple) -> Tuple: method _slice_tuple_from_tuple (line 174) | def _slice_tuple_from_tuple(st: Tuple) -> Tuple: method _prox_core (line 185) | def _prox_core( method prox (line 208) | def prox(self, v: Array, lam: float = 1.0, **kwargs) -> Array: class AnisotropicTVNorm (line 239) | class AnisotropicTVNorm(TVNorm): method __init__ (line 270) | def __init__( class IsotropicTVNorm (line 297) | class IsotropicTVNorm(TVNorm): method __init__ (line 328) | def __init__( class SingleAxisFiniteSum (line 355) | class SingleAxisFiniteSum(LinearOperator): method __init__ (line 373) | def __init__( method _eval (line 414) | def _eval(self, x: snp.Array) -> snp.Array: class FiniteSum (line 418) | class FiniteSum(VerticalStack): method __init__ (line 426) | def __init__( class SingleAxisHaarTransform (line 458) | class SingleAxisHaarTransform(VerticalStack): method __init__ (line 468) | def __init__( class HaarTransform (line 499) | class HaarTransform(VerticalStack): method __init__ (line 508) | def __init__( FILE: scico/linop/_circconv.py class CircularConvolve (line 25) | class CircularConvolve(LinearOperator): method __init__ (line 77) | def __init__( method _dft_center_shift (line 154) | def _dft_center_shift(self, input_shape) -> np.ndarray: method _eval (line 187) | def _eval(self, x: snp.Array) -> snp.Array: method _adj (line 198) | def _adj(self, x: snp.Array) -> snp.Array: # type: ignore method __add__ (line 211) | def __add__(self, other): method __sub__ (line 224) | def __sub__(self, other): method __mul__ (line 237) | def __mul__(self, scalar): method __truediv__ (line 247) | def __truediv__(self, scalar): method from_operator (line 257) | def from_operator( FILE: scico/linop/_convolve.py class Convolve (line 25) | class Convolve(LinearOperator): method __init__ (line 28) | def __init__( method _eval (line 78) | def _eval(self, x: snp.Array) -> snp.Array: method __add__ (line 82) | def __add__(self, other): method __sub__ (line 99) | def __sub__(self, other): method __mul__ (line 115) | def __mul__(self, scalar): method __truediv__ (line 126) | def __truediv__(self, scalar): class ConvolveByX (line 137) | class ConvolveByX(LinearOperator): method __init__ (line 144) | def __init__( method _eval (line 198) | def _eval(self, h: snp.Array) -> snp.Array: method __add__ (line 202) | def __add__(self, other): method __sub__ (line 217) | def __sub__(self, other): method __mul__ (line 234) | def __mul__(self, scalar): method __truediv__ (line 245) | def __truediv__(self, scalar): FILE: scico/linop/_dft.py class DFT (line 24) | class DFT(LinearOperator): method __init__ (line 27) | def __init__( method _eval (line 88) | def _eval(self, x: snp.Array) -> snp.Array: method inv (line 91) | def inv(self, z: snp.Array) -> snp.Array: FILE: scico/linop/_diag.py class Diagonal (line 26) | class Diagonal(LinearOperator): method __init__ (line 29) | def __init__( method _eval (line 70) | def _eval(self, x: Union[Array, BlockArray]) -> Union[Array, BlockArray]: method diagonal (line 74) | def diagonal(self) -> Union[Array, BlockArray]: method T (line 79) | def T(self) -> Diagonal: method conj (line 83) | def conj(self) -> Diagonal: method H (line 88) | def H(self) -> Diagonal: method gram_op (line 93) | def gram_op(self) -> Diagonal: method __add__ (line 102) | def __add__(self, other): method __sub__ (line 108) | def __sub__(self, other): method __mul__ (line 114) | def __mul__(self, scalar): method __truediv__ (line 118) | def __truediv__(self, scalar): method __matmul__ (line 121) | def __matmul__(self, other): method norm (line 130) | def norm(self, ord=None): # pylint: disable=W0622 class ScaledIdentity (line 155) | class ScaledIdentity(Diagonal): method __init__ (line 158) | def __init__( method diagonal (line 183) | def diagonal(self) -> Union[Array, BlockArray]: method conj (line 186) | def conj(self) -> ScaledIdentity: method gram_op (line 193) | def gram_op(self) -> ScaledIdentity: method __add__ (line 202) | def __add__(self, other): method __sub__ (line 212) | def __sub__(self, other): method __mul__ (line 222) | def __mul__(self, scalar): method __truediv__ (line 230) | def __truediv__(self, scalar): method __matmul__ (line 237) | def __matmul__(self, other): method norm (line 253) | def norm(self, ord=None): # pylint: disable=W0622 class Identity (line 271) | class Identity(ScaledIdentity): method __init__ (line 274) | def __init__( method _eval (line 289) | def _eval(self, x: Union[Array, BlockArray]) -> Union[Array, BlockArray]: method conj (line 292) | def conj(self) -> Identity: method gram_op (line 297) | def gram_op(self) -> Identity: method __matmul__ (line 301) | def __matmul__(self, other): method __rmatmul__ (line 304) | def __rmatmul__(self, x: Union[Array, BlockArray]) -> Union[Array, Blo... FILE: scico/linop/_diff.py class FiniteDifference (line 25) | class FiniteDifference(VerticalStack): method __init__ (line 47) | def __init__( class SingleAxisFiniteDifference (line 99) | class SingleAxisFiniteDifference(LinearOperator): method __init__ (line 157) | def __init__( method _eval (line 236) | def _eval(self, x: snp.Array) -> snp.Array: FILE: scico/linop/_func.py function linop_from_function (line 23) | def linop_from_function(f: Callable, classname: str, f_name: Optional[st... class Crop (line 97) | class Crop(LinearOperator): method __init__ (line 100) | def __init__( class Slice (line 141) | class Slice(LinearOperator): method __init__ (line 144) | def __init__( method _eval (line 184) | def _eval(self, x: snp.Array) -> snp.Array: FILE: scico/linop/_grad.py function diffstack (line 25) | def diffstack(x: Array, axis: Optional[Union[int, Tuple[int, ...]]] = No... class ProjectedGradient (line 51) | class ProjectedGradient(LinearOperator): method __init__ (line 80) | def __init__( method _eval (line 162) | def _eval(self, x: Array) -> Union[Array, BlockArray]: class PolarGradient (line 184) | class PolarGradient(ProjectedGradient): method __init__ (line 204) | def __init__( class CylindricalGradient (line 281) | class CylindricalGradient(ProjectedGradient): method __init__ (line 301) | def __init__( class SphericalGradient (line 407) | class SphericalGradient(ProjectedGradient): method __init__ (line 427) | def __init__( FILE: scico/linop/_linop.py function _wrap_add_sub (line 30) | def _wrap_add_sub(func: Callable) -> Callable: class LinearOperator (line 123) | class LinearOperator(Operator): method __init__ (line 126) | def __init__( method _set_adjoint (line 187) | def _set_adjoint(self): method _set_gram (line 194) | def _set_gram(self): method jit (line 198) | def jit(self): method __add__ (line 213) | def __add__(self, other): method __sub__ (line 224) | def __sub__(self, other): method __mul__ (line 235) | def __mul__(self, other): method __rmul__ (line 246) | def __rmul__(self, other): method __truediv__ (line 250) | def __truediv__(self, other): method __matmul__ (line 260) | def __matmul__(self, other): method __rmatmul__ (line 264) | def __rmatmul__(self, other): method __call__ (line 279) | def __call__( method adj (line 296) | def adj( method T (line 330) | def T(self) -> LinearOperator: method H (line 363) | def H(self) -> LinearOperator: method conj (line 387) | def conj(self) -> LinearOperator: method gram_op (line 404) | def gram_op(self) -> LinearOperator: method gram (line 422) | def gram( class ComposedLinearOperator (line 443) | class ComposedLinearOperator(LinearOperator): method __init__ (line 450) | def __init__(self, A: LinearOperator, B: LinearOperator, jit: bool = F... FILE: scico/linop/_matrix.py function _wrap_add_sub_matrix (line 29) | def _wrap_add_sub_matrix(func, op): class MatrixOperator (line 64) | class MatrixOperator(LinearOperator): method __init__ (line 67) | def __init__(self, A: ArrayLike, input_cols: int = 0): method __call__ (line 103) | def __call__(self, other): method _eval (line 127) | def _eval(self, other): method gram (line 130) | def gram(self, other): method __add__ (line 134) | def __add__(self, other): method __sub__ (line 138) | def __sub__(self, other): method __radd__ (line 141) | def __radd__(self, other): method __rsub__ (line 145) | def __rsub__(self, other): method __neg__ (line 148) | def __neg__(self): method __mul__ (line 153) | def __mul__(self, other): method __rmul__ (line 172) | def __rmul__(self, other): method __truediv__ (line 176) | def __truediv__(self, other): method __rtruediv__ (line 195) | def __rtruediv__(self, other): method __getitem__ (line 209) | def __getitem__(self, key): method T (line 213) | def T(self): method H (line 222) | def H(self): method conj (line 230) | def conj(self): method adj (line 238) | def adj(self, y): method to_array (line 241) | def to_array(self): method gram_op (line 246) | def gram_op(self): method norm (line 253) | def norm(self, ord=None, axis=None, keepdims=False): # pylint: disabl... FILE: scico/linop/_stack.py class VerticalStack (line 25) | class VerticalStack(VerticalStackOperator, LinearOperator): method __init__ (line 50) | def __init__( method _adj (line 71) | def _adj(self, y: Union[Array, BlockArray]) -> Array: # type: ignore class DiagonalStack (line 75) | class DiagonalStack(DiagonalStackOperator, LinearOperator): method __init__ (line 114) | def __init__( method _adj (line 143) | def _adj(self, y: Union[Array, BlockArray]) -> Union[Array, BlockArray... class DiagonalReplicated (line 150) | class DiagonalReplicated(DiagonalReplicatedOperator, LinearOperator): method __init__ (line 190) | def __init__( function linop_over_axes (line 227) | def linop_over_axes( FILE: scico/linop/_util.py function power_iteration (line 23) | def power_iteration(A: LinearOperator, maxiter: int = 100, key: Optional... function operator_norm (line 58) | def operator_norm(A: LinearOperator, maxiter: int = 100, key: Optional[P... function valid_adjoint (line 91) | def valid_adjoint( function jacobian (line 164) | def jacobian(F: Operator, u: snp.Array, include_eval: Optional[bool] = F... FILE: scico/linop/optics.py function _isscalar (line 69) | def _isscalar(element: Any) -> TypeGuard[Union[int, float]]: function radial_transverse_frequency (line 74) | def radial_transverse_frequency( class Propagator (line 120) | class Propagator(LinearOperator): method __init__ (line 123) | def __init__( method __repr__ (line 193) | def __repr__(self): method _eval (line 202) | def _eval(self, x): class AngularSpectrumPropagator (line 206) | class AngularSpectrumPropagator(Propagator): method __init__ (line 253) | def __init__( method adequate_sampling (line 293) | def adequate_sampling(self): method pinv (line 314) | def pinv(self, y): class FresnelPropagator (line 320) | class FresnelPropagator(Propagator): method __init__ (line 367) | def __init__( method adequate_sampling (line 389) | def adequate_sampling(self): class FraunhoferPropagator (line 410) | class FraunhoferPropagator(LinearOperator): method __init__ (line 464) | def __init__( method __repr__ (line 546) | def __repr__(self): method _eval (line 557) | def _eval(self, x): method adequate_sampling (line 563) | def adequate_sampling(self): FILE: scico/linop/xray/_axitom/backprojection.py function map_object_to_detector_coords (line 22) | def map_object_to_detector_coords(object_xs, object_ys, object_zs, config): function _fdk_axisym (line 76) | def _fdk_axisym(projection_filtered, config, angles): function fdk (line 133) | def fdk(projection: Array, config: Config, angles: Optional[Array] = Non... FILE: scico/linop/xray/_axitom/config.py class Config (line 14) | class Config: method __init__ (line 17) | def __init__( method __repr__ (line 86) | def __repr__(self): method with_param (line 98) | def with_param(self, **kwargs): FILE: scico/linop/xray/_axitom/filtering.py function _ramp_kernel_real (line 16) | def _ramp_kernel_real(cutoff, length): function _add_weights (line 36) | def _add_weights(projection, config): function ramp_filter_and_weight (line 62) | def ramp_filter_and_weight(projection, config): FILE: scico/linop/xray/_axitom/projection.py function _partial_forward_project (line 24) | def _partial_forward_project( function forward_project (line 82) | def forward_project( FILE: scico/linop/xray/_axitom/utilities.py function _find_center_of_gravity_in_projection (line 16) | def _find_center_of_gravity_in_projection(projection, background_interns... function find_center_of_rotation (line 60) | def find_center_of_rotation(projection, background_internsity=0.9, metho... function rotate_coordinates (line 94) | def rotate_coordinates(xs_array, ys_array, angle_rad): FILE: scico/linop/xray/_util.py function image_centroid (line 33) | def image_centroid(v: ArrayLike, center_offset: bool = False) -> Tuple[f... function center_image (line 61) | def center_image( function rotate_volume (line 89) | def rotate_volume( function image_alignment_rotation (line 132) | def image_alignment_rotation( function volume_alignment_rotation (line 206) | def volume_alignment_rotation( FILE: scico/linop/xray/_xray.py class XRayTransform2D (line 28) | class XRayTransform2D(LinearOperator): method __init__ (line 51) | def __init__( method project (line 128) | def project(self, im: ArrayLike) -> snp.Array: method back_project (line 136) | def back_project(self, y: ArrayLike) -> snp.Array: method fbp (line 144) | def fbp(self, y: ArrayLike) -> snp.Array: method _ramp_filter (line 186) | def _ramp_filter(x: ArrayLike, tau: float) -> snp.Array: method _project (line 211) | def _project( method _back_project (line 247) | def _back_project( method _calc_weights (line 280) | def _calc_weights( class XRayTransform3D (line 323) | class XRayTransform3D(LinearOperator): method __init__ (line 352) | def __init__( method project (line 376) | def project(self, im: ArrayLike) -> snp.Array: method back_project (line 380) | def back_project(self, proj: ArrayLike) -> snp.Array: method _project (line 385) | def _project(im: ArrayLike, matrices: ArrayLike, det_shape: Shape) -> ... method _project_single (line 411) | def _project_single( method _back_project (line 431) | def _back_project(proj: ArrayLike, matrices: ArrayLike, input_shape: S... method _back_project_single (line 458) | def _back_project_single( method _calc_weights (line 471) | def _calc_weights( method matrices_from_euler_angles (line 533) | def matrices_from_euler_angles( FILE: scico/linop/xray/abel.py class AbelTransform (line 30) | class AbelTransform(LinearOperator): method __init__ (line 38) | def __init__(self, img_shape: Shape): method _eval (line 54) | def _eval(self, x: jax.Array) -> jax.Array: method _adj (line 59) | def _adj(self, x: jax.Array) -> jax.Array: # type: ignore method inverse (line 64) | def inverse(self, y: jax.Array) -> jax.Array: function _pyabel_transform (line 78) | def _pyabel_transform( function _pyabel_daun_get_proj_matrix (line 120) | def _pyabel_daun_get_proj_matrix(img_shape: Shape) -> jax.Array: FILE: scico/linop/xray/astra.py function set_astra_gpu_index (line 66) | def set_astra_gpu_index(idx: Union[int, Sequence[int]]): function _project_coords (line 75) | def _project_coords( function project_world_coordinates (line 103) | def project_world_coordinates( function volume_coords_to_world_coords (line 140) | def volume_coords_to_world_coords(idx: np.ndarray, vol_geom: VolumeGeome... function _volume_index_to_astra_world_2d (line 160) | def _volume_index_to_astra_world_2d(idx: np.ndarray, vol_geom: VolumeGeo... function _volume_index_to_astra_world_3d (line 180) | def _volume_index_to_astra_world_3d(idx: np.ndarray, vol_geom: VolumeGeo... class XRayTransform2D (line 202) | class XRayTransform2D(LinearOperator): method __init__ (line 210) | def __init__( method _proj (line 304) | def _proj(self, x: jax.Array) -> jax.Array: method _bproj (line 315) | def _bproj(self, y: jax.Array) -> jax.Array: method fbp (line 325) | def fbp(self, sino: jax.Array, filter_type: str = "Ram-Lak") -> jax.Ar... function convert_from_scico_geometry (line 371) | def convert_from_scico_geometry( function _astra_to_scico_geometry (line 417) | def _astra_to_scico_geometry(vol_geom: VolumeGeometry, proj_geom: Projec... function convert_to_scico_geometry (line 448) | def convert_to_scico_geometry( class XRayTransform3D (line 488) | class XRayTransform3D(LinearOperator): # pragma: no cover method __init__ (line 580) | def __init__( method create_astra_geometry (line 674) | def create_astra_geometry( method _proj (line 720) | def _proj(self, x: jax.Array) -> jax.Array: method _bproj (line 731) | def _bproj(self, y: jax.Array) -> jax.Array: function angle_to_vector (line 742) | def angle_to_vector(det_spacing: Tuple[float, float], angles: np.ndarray... function rotate_vectors (line 763) | def rotate_vectors(vectors: np.ndarray, rot: Rotation) -> np.ndarray: function _ensure_writeable (line 781) | def _ensure_writeable(x): FILE: scico/linop/xray/svmbir.py class XRayTransform (line 35) | class XRayTransform(LinearOperator): method __init__ (line 66) | def __init__( method _proj (line 205) | def _proj( method _proj_hcb (line 233) | def _proj_hcb(self, x): method _bproj (line 255) | def _bproj( method _bproj_hcb (line 285) | def _bproj_hcb(self, y): class SVMBIRExtendedLoss (line 308) | class SVMBIRExtendedLoss(Loss): method __init__ (line 336) | def __init__( method __call__ (line 391) | def __call__(self, x: snp.Array) -> float: method prox (line 397) | def prox(self, v: snp.Array, lam: float = 1, **kwargs) -> snp.Array: class SVMBIRSquaredL2Loss (line 437) | class SVMBIRSquaredL2Loss(SVMBIRExtendedLoss, SquaredL2Loss): method __init__ (line 453) | def __init__( FILE: scico/linop/xray/symcone.py function _volume_by_axial_symmetry (line 34) | def _volume_by_axial_symmetry( class AxiallySymmetricVolume (line 83) | class AxiallySymmetricVolume(LinearOperator): method __init__ (line 86) | def __init__( class SymConeXRayTransform (line 115) | class SymConeXRayTransform(LinearOperator): method __init__ (line 125) | def __init__( method fdk (line 177) | def fdk(self, y: ArrayLike, num_angles: int = 360): FILE: scico/loss.py function _loss_mul_div_wrapper (line 27) | def _loss_mul_div_wrapper(func): class Loss (line 40) | class Loss(functional.Functional): method __init__ (line 52) | def __init__( method __call__ (line 87) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 102) | def prox( method __mul__ (line 133) | def __mul__(self, other): method __rmul__ (line 139) | def __rmul__(self, other): method __truediv__ (line 143) | def __truediv__(self, other): method set_scale (line 149) | def set_scale(self, new_scale: float): class SquaredL2Loss (line 154) | class SquaredL2Loss(Loss): method __init__ (line 170) | def __init__( method __call__ (line 208) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 211) | def prox( method hessian (line 246) | def hessian(self) -> linop.LinearOperator: class PoissonLoss (line 270) | class PoissonLoss(Loss): method __init__ (line 282) | def __init__( method __call__ (line 300) | def __call__(self, x: Union[Array, BlockArray]) -> float: class SquaredL2AbsLoss (line 305) | class SquaredL2AbsLoss(Loss): method __init__ (line 324) | def __init__( method __call__ (line 354) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 357) | def prox( function _cbrt (line 371) | def _cbrt(x: Union[Array, BlockArray]) -> Union[Array, BlockArray]: function _check_root (line 391) | def _check_root( function _dep_cubic_root (line 418) | def _dep_cubic_root( class SquaredL2SquaredAbsLoss (line 509) | class SquaredL2SquaredAbsLoss(Loss): method __init__ (line 528) | def __init__( method __call__ (line 558) | def __call__(self, x: Union[Array, BlockArray]) -> float: method prox (line 563) | def prox( FILE: scico/metric.py function mae (line 20) | def mae(reference: Union[Array, BlockArray], comparison: Union[Array, Bl... function mse (line 34) | def mse(reference: Union[Array, BlockArray], comparison: Union[Array, Bl... function snr (line 48) | def snr(reference: Union[Array, BlockArray], comparison: Union[Array, Bl... function psnr (line 65) | def psnr( function isnr (line 95) | def isnr( function bsnr (line 121) | def bsnr(blurry: Union[Array, BlockArray], noisy: Union[Array, BlockArra... function rel_res (line 142) | def rel_res(ax: Union[BlockArray, Array], b: Union[BlockArray, Array]) -... FILE: scico/numpy/__init__.py function ravel (line 63) | def ravel(ba: Union[Array | BlockArray]) -> Array: FILE: scico/numpy/_blockarray.py class BlockArray (line 25) | class BlockArray: method __init__ (line 52) | def __init__(self, inputs): method dtype (line 61) | def dtype(self): method __len__ (line 69) | def __len__(self): method __getitem__ (line 72) | def __getitem__(self, key): method __setitem__ (line 83) | def __setitem__(self, key, value): method blockarray (line 87) | def blockarray(iterable): method __repr__ (line 91) | def __repr__(self): method stack (line 94) | def stack(self, axis=0): function _unary_op_wrapper (line 126) | def _unary_op_wrapper(op_name): function _binary_op_wrapper (line 141) | def _binary_op_wrapper(op_name): function _jax_array_prop_wrapper (line 166) | def _jax_array_prop_wrapper(prop_name): function _jax_array_method_wrapper (line 197) | def _jax_array_method_wrapper(method_name): FILE: scico/numpy/_wrappers.py function add_attributes (line 25) | def add_attributes( function wrap_recursively (line 56) | def wrap_recursively( function map_func_over_args (line 74) | def map_func_over_args( function add_full_reduction (line 177) | def add_full_reduction(func: Callable, axis_arg_name: Optional[str] = "a... FILE: scico/numpy/util.py function transpose_ntpl_of_list (line 26) | def transpose_ntpl_of_list(ntpl: NamedTuple) -> List[NamedTuple]: function transpose_list_of_ntpl (line 41) | def transpose_list_of_ntpl(ntlist: List[NamedTuple]) -> NamedTuple: function namedtuple_to_array (line 56) | def namedtuple_to_array(ntpl: NamedTuple) -> snp.Array: function array_to_namedtuple (line 78) | def array_to_namedtuple(array: snp.Array) -> NamedTuple: function normalize_axes (line 97) | def normalize_axes( function slice_length (line 154) | def slice_length(length: int, idx: AxisIndex) -> Optional[int]: function indexed_shape (line 190) | def indexed_shape(shape: Shape, idx: ArrayIndex) -> Tuple[int, ...]: function jax_indexed_shape (line 228) | def jax_indexed_shape(shape: Shape, idx: ArrayIndex) -> Tuple[int, ...]: function no_nan_divide (line 268) | def no_nan_divide( function _readable_size (line 284) | def _readable_size(size: int) -> str: function array_info (line 305) | def array_info(x: Union[snp.BlockArray, snp.Array]) -> str: function shape_to_size (line 340) | def shape_to_size(shape: Union[Shape, BlockShape]) -> int: function is_array (line 360) | def is_array(x: Any) -> bool: function is_arraylike (line 375) | def is_arraylike(x: Any) -> bool: function is_nested (line 390) | def is_nested(x: Any) -> bool: function is_collapsible (line 412) | def is_collapsible(shapes: Sequence[Union[Shape, BlockShape]]) -> bool: function is_blockable (line 427) | def is_blockable(shapes: Sequence[Union[Shape, BlockShape]]) -> TypeGuar... function shape_dtype_rep (line 442) | def shape_dtype_rep( function broadcast_nested_shapes (line 464) | def broadcast_nested_shapes( function is_real_dtype (line 500) | def is_real_dtype(dtype: DType) -> bool: function is_complex_dtype (line 513) | def is_complex_dtype(dtype: DType) -> bool: function real_dtype (line 526) | def real_dtype(dtype: DType) -> DType: function complex_dtype (line 544) | def complex_dtype(dtype: DType) -> DType: function dtype_name (line 562) | def dtype_name(dtype: DType) -> str: function is_scalar_equiv (line 578) | def is_scalar_equiv(s: Any) -> bool: FILE: scico/operator/_func.py function operator_from_function (line 25) | def operator_from_function(f: Callable, classname: str, f_name: Optional... FILE: scico/operator/_operator.py function _wrap_mul_div_scalar (line 29) | def _wrap_mul_div_scalar(func: Callable) -> Callable: class Operator (line 63) | class Operator: method __init__ (line 69) | def __init__( method jit (line 172) | def jit(self): method __str__ (line 176) | def __str__(self): method __repr__ (line 179) | def __repr__(self): method __call__ (line 187) | def __call__(self, x: Union[Operator, Array, BlockArray]) -> Union[Ope... method __add__ (line 227) | def __add__(self, other: Operator) -> Operator: method __sub__ (line 240) | def __sub__(self, other: Operator) -> Operator: method __mul__ (line 254) | def __mul__(self, other): method __neg__ (line 263) | def __neg__(self) -> Operator: method __rmul__ (line 267) | def __rmul__(self, other): method __truediv__ (line 277) | def __truediv__(self, other): method jvp (line 286) | def jvp(self, u, v): method vjp (line 305) | def vjp(self, u, conjugate=True): method freeze (line 342) | def freeze(self, argnum: int, val: Union[Array, BlockArray]) -> Operator: FILE: scico/operator/_stack.py function collapse_shapes (line 26) | def collapse_shapes( class VerticalStack (line 46) | class VerticalStack(Operator): method __init__ (line 63) | def __init__( method check_if_stackable (line 102) | def check_if_stackable(ops: Sequence[Operator]): method _eval (line 126) | def _eval(self, x: Array) -> Union[Array, BlockArray]: method __repr__ (line 131) | def __repr__(self): class DiagonalStack (line 136) | class DiagonalStack(Operator): method __init__ (line 167) | def __init__( method check_if_stackable (line 208) | def check_if_stackable(ops: Sequence[Operator]): method _eval (line 220) | def _eval(self, x: Union[Array, BlockArray]) -> Union[Array, BlockArray]: method __repr__ (line 226) | def __repr__(self): class DiagonalReplicated (line 231) | class DiagonalReplicated(Operator): method __init__ (line 263) | def __init__( method __repr__ (line 334) | def __repr__(self): FILE: scico/operator/biconvolve.py class BiConvolve (line 28) | class BiConvolve(Operator): method __init__ (line 40) | def __init__( method _eval (line 81) | def _eval(self, x: BlockArray) -> Array: method freeze (line 84) | def freeze(self, argnum: int, val: Array) -> scico.linop.LinearOperator: FILE: scico/optimize/_admm.py class ADMM (line 33) | class ADMM(Optimizer): method __init__ (line 94) | def __init__( method _working_vars_finite (line 150) | def _working_vars_finite(self) -> bool: method _objective_evaluatable (line 167) | def _objective_evaluatable(self): method _itstat_extra_fields (line 171) | def _itstat_extra_fields(self): method _state_variable_names (line 200) | def _state_variable_names(self) -> List[str]: method minimizer (line 207) | def minimizer(self) -> Union[Array, BlockArray]: method objective (line 210) | def objective( method norm_primal_residual (line 254) | def norm_primal_residual(self, x: Optional[Union[Array, BlockArray]] =... method norm_dual_residual (line 279) | def norm_dual_residual(self) -> float: method z_init (line 297) | def z_init( method u_init (line 317) | def u_init(self, x0: Union[Array, BlockArray]) -> List[Union[Array, Bl... method step (line 334) | def step(self): FILE: scico/optimize/_admmaux.py class SubproblemSolver (line 39) | class SubproblemSolver: method internal_init (line 60) | def internal_init(self, admm: soa.ADMM): class GenericSubproblemSolver (line 70) | class GenericSubproblemSolver(SubproblemSolver): method __init__ (line 82) | def __init__(self, minimize_kwargs: dict = {"options": {"maxiter": 100... method solve (line 92) | def solve(self, x0: Union[Array, BlockArray]) -> Union[Array, BlockArr... class LinearSubproblemSolver (line 120) | class LinearSubproblemSolver(SubproblemSolver): method __init__ (line 162) | def __init__(self, cg_kwargs: Optional[dict[str, Any]] = None, cg_func... method internal_init (line 199) | def internal_init(self, admm: soa.ADMM): method compute_rhs (line 225) | def compute_rhs(self) -> Union[Array, BlockArray]: method solve (line 252) | def solve(self, x0: Union[Array, BlockArray]) -> Union[Array, BlockArr... class MatrixSubproblemSolver (line 266) | class MatrixSubproblemSolver(LinearSubproblemSolver): method __init__ (line 305) | def __init__(self, check_solve: bool = False, solve_kwargs: Optional[d... method internal_init (line 320) | def internal_init(self, admm: soa.ADMM): method solve (line 353) | def solve(self, x0: Array) -> Array: class CircularConvolveSolver (line 370) | class CircularConvolveSolver(LinearSubproblemSolver): method __init__ (line 391) | def __init__(self, ndims: Optional[int] = None): method internal_init (line 406) | def internal_init(self, admm: soa.ADMM): method solve (line 449) | def solve(self, x0: Union[Array, BlockArray]) -> Union[Array, BlockArr... class FBlockCircularConvolveSolver (line 468) | class FBlockCircularConvolveSolver(LinearSubproblemSolver): method __init__ (line 568) | def __init__(self, ndims: Optional[int] = None, check_solve: bool = Fa... method internal_init (line 579) | def internal_init(self, admm: soa.ADMM): method solve (line 608) | def solve(self, x0: Union[Array, BlockArray]) -> Union[Array, BlockArr... class G0BlockCircularConvolveSolver (line 627) | class G0BlockCircularConvolveSolver(SubproblemSolver): method __init__ (line 739) | def __init__(self, ndims: Optional[int] = None, check_solve: bool = Fa... method internal_init (line 750) | def internal_init(self, admm: soa.ADMM): method compute_rhs (line 782) | def compute_rhs(self) -> Union[Array, BlockArray]: method solve (line 812) | def solve(self, x0: Union[Array, BlockArray]) -> Union[Array, BlockArr... FILE: scico/optimize/_common.py function itstat_func_and_object (line 28) | def itstat_func_and_object( class Optimizer (line 77) | class Optimizer: method __init__ (line 86) | def __init__(self, **kwargs: Any): method _working_vars_finite (line 137) | def _working_vars_finite(self) -> bool: method _itstat_default_fields (line 147) | def _itstat_default_fields(self) -> Tuple[Dict[str, str], List[str]]: method _objective_evaluatable (line 167) | def _objective_evaluatable(self) -> bool: method _itstat_extra_fields (line 175) | def _itstat_extra_fields(self) -> Tuple[Dict[str, str], List[str]]: method _state_variable_names (line 186) | def _state_variable_names(self) -> List[str]: method _get_state_variables (line 197) | def _get_state_variables(self) -> dict[str, Any]: method _set_state_variables (line 207) | def _set_state_variables(self, **kwargs): method save_state (line 222) | def save_state(self, path: str): method load_state (line 239) | def load_state(self, path: str): method history (line 260) | def history(self, transpose: bool = False) -> Union[List[NamedTuple], ... method minimizer (line 275) | def minimizer(self) -> Union[Array, BlockArray]: method step (line 283) | def step(self): method solve (line 287) | def solve( FILE: scico/optimize/_ladmm.py class LinearizedADMM (line 25) | class LinearizedADMM(Optimizer): method __init__ (line 84) | def __init__( method _working_vars_finite (line 123) | def _working_vars_finite(self) -> bool: method _objective_evaluatable (line 135) | def _objective_evaluatable(self): method _itstat_extra_fields (line 139) | def _itstat_extra_fields(self): method _state_variable_names (line 145) | def _state_variable_names(self) -> List[str]: method minimizer (line 148) | def minimizer(self) -> Union[Array, BlockArray]: method objective (line 151) | def objective( method norm_primal_residual (line 183) | def norm_primal_residual(self, x: Optional[Union[Array, BlockArray]] =... method norm_dual_residual (line 204) | def norm_dual_residual(self) -> float: method z_init (line 217) | def z_init( method u_init (line 236) | def u_init(self, x0: Union[Array, BlockArray]) -> Union[Array, BlockAr... method step (line 253) | def step(self): FILE: scico/optimize/_padmm.py class ProximalADMMBase (line 30) | class ProximalADMMBase(Optimizer): method __init__ (line 51) | def __init__( method _working_vars_finite (line 118) | def _working_vars_finite(self) -> bool: method _objective_evaluatable (line 130) | def _objective_evaluatable(self): method _itstat_extra_fields (line 134) | def _itstat_extra_fields(self): method _state_variable_names (line 140) | def _state_variable_names(self) -> List[str]: method minimizer (line 143) | def minimizer(self) -> Union[Array, BlockArray]: method objective (line 146) | def objective( class ProximalADMM (line 177) | class ProximalADMM(ProximalADMMBase): method __init__ (line 222) | def __init__( method norm_primal_residual (line 291) | def norm_primal_residual( method norm_dual_residual (line 322) | def norm_dual_residual(self) -> float: method step (line 346) | def step(self): method estimate_parameters (line 362) | def estimate_parameters( class NonLinearPADMM (line 407) | class NonLinearPADMM(ProximalADMMBase): method __init__ (line 453) | def __init__( method norm_primal_residual (line 509) | def norm_primal_residual( method norm_dual_residual (line 540) | def norm_dual_residual(self) -> float: method step (line 574) | def step(self): method estimate_parameters (line 590) | def estimate_parameters( FILE: scico/optimize/_pgm.py class PGM (line 33) | class PGM(Optimizer): method __init__ (line 65) | def __init__( method x_step (line 109) | def x_step(self, v: Union[Array, BlockArray], L: float) -> Union[Array... method _x_step (line 115) | def _x_step( method _working_vars_finite (line 121) | def _working_vars_finite(self) -> bool: method _objective_evaluatable (line 129) | def _objective_evaluatable(self): method _itstat_extra_fields (line 133) | def _itstat_extra_fields(self): method _state_variable_names (line 139) | def _state_variable_names(self) -> List[str]: method minimizer (line 142) | def minimizer(self) -> Union[Array, BlockArray]: method objective (line 145) | def objective(self, x: Optional[Union[Array, BlockArray]] = None) -> f... method f_quad_approx (line 151) | def f_quad_approx( method norm_residual (line 168) | def norm_residual(self) -> float: method step (line 176) | def step(self): class AcceleratedPGM (line 185) | class AcceleratedPGM(PGM): method __init__ (line 196) | def __init__( method step (line 223) | def step(self): method _working_vars_finite (line 244) | def _working_vars_finite(self) -> bool: method _state_variable_names (line 252) | def _state_variable_names(self) -> List[str]: FILE: scico/optimize/_pgmaux.py class PGMStepSize (line 23) | class PGMStepSize: method internal_init (line 40) | def internal_init(self, pgm: sop.PGM): method update (line 49) | def update(self, v: Union[Array, BlockArray]) -> float: class BBStepSize (line 65) | class BBStepSize(PGMStepSize): method __init__ (line 89) | def __init__(self): method update (line 94) | def update(self, v: Union[Array, BlockArray]) -> float: class AdaptiveBBStepSize (line 128) | class AdaptiveBBStepSize(PGMStepSize): method __init__ (line 169) | def __init__(self, kappa: float = 0.5): method update (line 181) | def update(self, v: Union[Array, BlockArray]) -> float: class LineSearchStepSize (line 232) | class LineSearchStepSize(PGMStepSize): method __init__ (line 254) | def __init__(self, gamma_u: float = 1.2, maxiter: int = 50): method update (line 269) | def update(self, v: Union[Array, BlockArray]) -> float: class RobustLineSearchStepSize (line 295) | class RobustLineSearchStepSize(LineSearchStepSize): method __init__ (line 319) | def __init__(self, gamma_d: float = 0.9, gamma_u: float = 2.0, maxiter... method update (line 335) | def update(self, v: Union[Array, BlockArray]) -> float: FILE: scico/optimize/_primaldual.py class PDHG (line 27) | class PDHG(Optimizer): method __init__ (line 95) | def __init__( method _working_vars_finite (line 145) | def _working_vars_finite(self) -> bool: method _objective_evaluatable (line 153) | def _objective_evaluatable(self): method _itstat_extra_fields (line 157) | def _itstat_extra_fields(self): method _state_variable_names (line 163) | def _state_variable_names(self) -> List[str]: method minimizer (line 166) | def minimizer(self) -> Union[Array, BlockArray]: method objective (line 169) | def objective( method norm_primal_residual (line 192) | def norm_primal_residual(self) -> float: method norm_dual_residual (line 206) | def norm_dual_residual(self) -> float: method step (line 220) | def step(self): method estimate_parameters (line 235) | def estimate_parameters( FILE: scico/plot.py function _attach_keypress (line 48) | def _attach_keypress(fig, scaling=1.1): function _attach_zoom (line 86) | def _attach_zoom(ax, scaling=2.0): function plot (line 180) | def plot(y, x=None, ptyp="plot", xlbl=None, ylbl=None, title=None, lgnd=... function surf (line 289) | def surf( function contour (line 406) | def contour( function imview (line 557) | def imview( function close (line 729) | def close(fig=None): function _in_ipython (line 746) | def _in_ipython(): function _in_notebook (line 762) | def _in_notebook(): function set_ipython_plot_backend (line 778) | def set_ipython_plot_backend(backend="qt"): function set_notebook_plot_backend (line 798) | def set_notebook_plot_backend(backend="inline"): function config_notebook_plotting (line 818) | def config_notebook_plotting(): FILE: scico/random.py function _add_seed (line 65) | def _add_seed(fun): function _wrap (line 129) | def _wrap(fun): function _is_wrappable (line 135) | def _is_wrappable(fun): function randn (line 149) | def randn( FILE: scico/ray/tune.py class _CustomReporter (line 41) | class _CustomReporter(TuneReporterBase): method should_report (line 44) | def should_report(self, trials: List[Trial], done: bool = False): method report (line 49) | def report(self, trials: List[Trial], done: bool, *sys_info: Dict): function run (line 76) | def run( class Tuner (line 187) | class Tuner(ray.tune.Tuner): method __init__ (line 190) | def __init__( method fit (line 310) | def fit(self) -> ResultGrid: FILE: scico/solver.py function _wrap_func (line 74) | def _wrap_func(func: Callable, shape: Union[Shape, BlockShape], dtype: D... function _wrap_func_and_grad (line 103) | def _wrap_func_and_grad(func: Callable, shape: Union[Shape, BlockShape],... function _split_real_imag (line 135) | def _split_real_imag(x: Union[Array, BlockArray]) -> Union[Array, BlockA... function _join_real_imag (line 154) | def _join_real_imag(x: Union[Array, BlockArray]) -> Union[Array, BlockAr... function _ravel (line 172) | def _ravel(x: Union[Array, BlockArray]) -> Array: function _unravel (line 187) | def _unravel(x: Array, shape: Union[Shape, BlockShape]) -> Union[Array, ... function minimize (line 206) | def minimize( function minimize_scalar (line 290) | def minimize_scalar( function cg (line 326) | def cg( function lstsq (line 408) | def lstsq( function bisect (line 469) | def bisect( function golden (line 543) | def golden( class MatrixATADSolver (line 613) | class MatrixATADSolver: method __init__ (line 726) | def __init__( method solve (line 805) | def solve(self, b: Array, check_finite: Optional[bool] = None) -> Array: method accuracy (line 840) | def accuracy(self, x: Array, b: Array) -> float: class ConvATADSolver (line 858) | class ConvATADSolver: method __init__ (line 940) | def __init__(self, A: ComposedLinearOperator, D: CircularConvolve): method solve (line 972) | def solve(self, b: Array) -> Array: method accuracy (line 997) | def accuracy(self, x: Array, b: Array) -> float: FILE: scico/test/conftest.py function pytest_addoption (line 8) | def pytest_addoption(parser, pluginmanager): function pytest_configure (line 22) | def pytest_configure(config): function pytest_collection_modifyitems (line 27) | def pytest_collection_modifyitems(config, items): FILE: scico/test/flax/test_apply.py function testobj (line 21) | def testobj(): function test_apply_fn (line 25) | def test_apply_fn(testobj): function test_except_only_apply (line 48) | def test_except_only_apply(testobj): function test_eval (line 62) | def test_eval(testobj, model_cls): function test_apply_from_checkpoint (line 87) | def test_apply_from_checkpoint(testobj): FILE: scico/test/flax/test_checkpoints.py function testobj (line 19) | def testobj(): function test_checkpoint (line 24) | def test_checkpoint(testobj): function test_checkpointing_from_trainer (line 66) | def test_checkpointing_from_trainer(testobj, model_cls): function test_checkpoint_exception (line 115) | def test_checkpoint_exception(testobj): FILE: scico/test/flax/test_clu.py function test_count_parameters (line 14) | def test_count_parameters(): function test_count_parameters_empty (line 54) | def test_count_parameters_empty(): function test_get_parameter_overview_empty (line 91) | def test_get_parameter_overview_empty(): class CNN (line 95) | class CNN(Module): method __call__ (line 97) | def __call__(self, x): function test_get_parameter_overview (line 102) | def test_get_parameter_overview(): function test_printing_bool (line 116) | def test_printing_bool(): FILE: scico/test/flax/test_examples_flax.py function test_foam1_gen (line 42) | def test_foam1_gen(): function test_foam2_gen (line 52) | def test_foam2_gen(): function test_distdatagen (line 62) | def test_distdatagen(): function test_ct_data_generation (line 80) | def test_ct_data_generation(): function test_blur_data_generation (line 97) | def test_blur_data_generation(): function test_rotation90 (line 113) | def test_rotation90(): function test_flip (line 124) | def test_flip(): function test_center_crop (line 136) | def test_center_crop(output_size): function test_positional_crop (line 154) | def test_positional_crop(output_size): function test_random_noise1 (line 170) | def test_random_noise1(range_flag): function test_random_noise2 (line 183) | def test_random_noise2(shape): function test_preprocess_images (line 194) | def test_preprocess_images(output_size, gray_flag, num_img_req): function test_preprocess_images_multi_flag (line 225) | def test_preprocess_images_multi_flag(): class SetupTest (line 251) | class SetupTest: method __init__ (line 252) | def __init__(self): function testobj (line 271) | def testobj(): function test_build_image_dataset (line 276) | def test_build_image_dataset(testobj, augment): function test_padded_circular_convolve (line 298) | def test_padded_circular_convolve(): function test_runtime_error_scalar (line 311) | def test_runtime_error_scalar(): function test_runtime_error_array (line 316) | def test_runtime_error_array(): function test_default_cache_path (line 321) | def test_default_cache_path(): function test_cache_path (line 331) | def test_cache_path(): FILE: scico/test/flax/test_flax.py class TestSet (line 17) | class TestSet: method test_convnblock_default (line 18) | def test_convnblock_default(self): method test_convnblock_args (line 31) | def test_convnblock_args(self): method test_convblock_default (line 49) | def test_convblock_default(self): method test_convblock_args (line 60) | def test_convblock_args(self): method test_convblock_call (line 76) | def test_convblock_call(self): method test_convnpblock_args (line 96) | def test_convnpblock_args(self): method test_convnublock_args (line 117) | def test_convnublock_args(self): method test_convmnblock_default (line 138) | def test_convmnblock_default(self): method test_upscale (line 153) | def test_upscale(self): method test_resnet_default (line 161) | def test_resnet_default(self): method test_unet_default (line 177) | def test_unet_default(self): class DnCNNNetTest (line 194) | class DnCNNNetTest: method __init__ (line 195) | def __init__(self): function testobj (line 210) | def testobj(): function test_DnCNN_call (line 214) | def test_DnCNN_call(testobj): function test_DnCNN_train (line 220) | def test_DnCNN_train(testobj): function test_DnCNN_test (line 240) | def test_DnCNN_test(testobj): function test_FlaxMap_call (line 250) | def test_FlaxMap_call(testobj): function test_FlaxMap_3D_call (line 261) | def test_FlaxMap_3D_call(testobj): function test_FlaxMap_batch_call (line 273) | def test_FlaxMap_batch_call(testobj): function test_FlaxMap_blockarray_exception (line 286) | def test_FlaxMap_blockarray_exception(testobj): function test_variable_load (line 301) | def test_variable_load(variant): function test_variable_load_mismatch (line 323) | def test_variable_load_mismatch(): function test_variable_save (line 339) | def test_variable_save(): FILE: scico/test/flax/test_inv.py class TestSet (line 19) | class TestSet: method setup_method (line 20) | def setup_method(self, method): method test_odpdn_default (line 27) | def test_odpdn_default(self): method test_odpdcnv_default (line 46) | def test_odpdcnv_default(self): method test_odpdcnv_padded (line 71) | def test_odpdcnv_padded(self): method test_train_odpdcnv_default (line 93) | def test_train_odpdcnv_default(self): class TestCT (line 150) | class TestCT: method setup_method (line 151) | def setup_method(self, method): method test_odpct_default (line 189) | def test_odpct_default(self): method test_modlct_default (line 206) | def test_modlct_default(self): method test_train_modl (line 222) | def test_train_modl(self): method test_train_odpct (line 254) | def test_train_odpct(self): FILE: scico/test/flax/test_spectral.py function test_l2_normalize (line 25) | def test_l2_normalize(): function test_conv (line 37) | def test_conv(kernel_size): class CNN (line 59) | class CNN(Module): method __call__ (line 64) | def __call__(self, x): function test_conv_layer (line 78) | def test_conv_layer(kernel_size): function test_spectral_norm (line 98) | def test_spectral_norm(input_shape): function test_spectral_norm_conv (line 112) | def test_spectral_norm_conv(kernel_shape): function test_train_spectral_norm (line 127) | def test_train_spectral_norm(): FILE: scico/test/flax/test_steps.py function testobj (line 19) | def testobj(): function test_basic_train_step (line 23) | def test_basic_train_step(testobj): function test_post_train_step (line 66) | def test_post_train_step(testobj): function test_basic_eval_step (line 114) | def test_basic_eval_step(testobj): FILE: scico/test/flax/test_train_aux.py function testobj (line 23) | def testobj(): function test_mse_loss (line 27) | def test_mse_loss(): function test_data_iterator (line 38) | def test_data_iterator(testobj, batch_size): function test_dstrain (line 46) | def test_dstrain(testobj, local_batch): function test_dstest (line 70) | def test_dstest(testobj, local_batch): function test_prepare_data (line 89) | def test_prepare_data(testobj): function test_compute_metrics (line 96) | def test_compute_metrics(testobj): function test_cnst_learning_rate (line 109) | def test_cnst_learning_rate(testobj): function test_cos_learning_rate (line 116) | def test_cos_learning_rate(testobj): function test_exp_learning_rate (line 130) | def test_exp_learning_rate(testobj): function test_train_initialize_function (line 143) | def test_train_initialize_function(testobj): function test_create_basic_train_state_default (line 171) | def test_create_basic_train_state_default(testobj): function test_create_basic_train_state (line 208) | def test_create_basic_train_state(testobj): function test_sgd_train_state (line 245) | def test_sgd_train_state(testobj): function test_sgd_no_momentum_train_state (line 270) | def test_sgd_no_momentum_train_state(testobj): function test_argument_struct (line 296) | def test_argument_struct(): function test_complete_stats_obj (line 309) | def test_complete_stats_obj(): function test_except_incomplete_stats_obj (line 332) | def test_except_incomplete_stats_obj(): function test_patch_incomplete_stats_obj (line 356) | def test_patch_incomplete_stats_obj(): FILE: scico/test/flax/test_trainer.py class SetupTest (line 21) | class SetupTest: method __init__ (line 22) | def __init__(self): function testobj (line 66) | def testobj(): function test_optimizers (line 71) | def test_optimizers(testobj, opt_type): function test_optimizers_exception (line 91) | def test_optimizers_exception(testobj): function test_sync_batch_stats (line 107) | def test_sync_batch_stats(testobj): function test_class_train_default_init (line 134) | def test_class_train_default_init(testobj): function test_class_train_default_noseed (line 152) | def test_class_train_default_noseed(testobj): function test_class_train_nolog (line 170) | def test_class_train_nolog(testobj): function test_class_train_required_steps (line 191) | def test_class_train_required_steps(testobj): function test_except_class_train_batch_size (line 218) | def test_except_class_train_batch_size(testobj): function test_class_train_set_steps (line 233) | def test_class_train_set_steps(testobj): function test_class_train_set_reporting (line 257) | def test_class_train_set_reporting(testobj): function test_class_train_set_functions (line 283) | def test_class_train_set_functions(testobj): function test_class_train_set_iterators (line 320) | def test_class_train_set_iterators(testobj): function test_class_train_set_parallel (line 341) | def test_class_train_set_parallel(testobj, postl): function test_class_train_external_init (line 371) | def test_class_train_external_init(testobj, chkflag): function test_class_train_train_loop (line 408) | def test_class_train_train_loop(testobj, model_cls): function test_class_train_train_post_loop (line 433) | def test_class_train_train_post_loop(testobj): function test_class_train_return_state (line 464) | def test_class_train_return_state(testobj): function test_class_train_update_metrics (line 487) | def test_class_train_update_metrics(testobj): function test_class_train_update_metrics_nolog (line 512) | def test_class_train_update_metrics_nolog(testobj): FILE: scico/test/flax/test_traversal.py function testobj (line 13) | def testobj(): function test_construct_traversal (line 18) | def test_construct_traversal(testobj, pname): FILE: scico/test/functional/prox.py function prox_func (line 7) | def prox_func(x, v, f, alpha): function prox_solve (line 14) | def prox_solve(v, v0, f, alpha): function prox_test (line 28) | def prox_test(v, nrm, prx, alpha, x0=None, rtol=1e-6): FILE: scico/test/functional/test_composed.py class TestComposed (line 14) | class TestComposed: method setup_method (line 15) | def setup_method(self): method test_eval (line 25) | def test_eval(self): method test_prox (line 28) | def test_prox(self): FILE: scico/test/functional/test_denoiser_func.py class TestBM3D (line 15) | class TestBM3D: method setup_method (line 16) | def setup_method(self): method test_gry (line 25) | def test_gry(self): method test_rgb (line 30) | def test_rgb(self): class TestBM4D (line 40) | class TestBM4D: method setup_method (line 41) | def setup_method(self): method test (line 48) | def test(self): class TestBlindDnCNN (line 54) | class TestBlindDnCNN: method setup_method (line 55) | def setup_method(self): method test_sngchn (line 62) | def test_sngchn(self): method test_mltchn (line 67) | def test_mltchn(self): class TestNonBlindDnCNN (line 73) | class TestNonBlindDnCNN: method setup_method (line 74) | def setup_method(self): method test_sngchn (line 81) | def test_sngchn(self): method test_mltchn (line 86) | def test_mltchn(self): FILE: scico/test/functional/test_funcional_core.py function pytest_generate_tests (line 28) | def pytest_generate_tests(metafunc): class ProxTestObj (line 44) | class ProxTestObj: method __init__ (line 45) | def __init__(self, dtype): function test_prox_obj (line 54) | def test_prox_obj(test_dtype): class SeparableTestObject (line 58) | class SeparableTestObject: method __init__ (line 59) | def __init__(self, dtype): function test_separable_obj (line 74) | def test_separable_obj(test_dtype): function test_separable_eval (line 78) | def test_separable_eval(test_separable_obj): function test_separable_prox (line 85) | def test_separable_prox(test_separable_obj): function test_separable_grad (line 94) | def test_separable_grad(test_separable_obj): class HuberNormSep (line 103) | class HuberNormSep(functional.HuberNorm): method __init__ (line 104) | def __init__(self, delta=1.0): class HuberNormNonSep (line 108) | class HuberNormNonSep(functional.HuberNorm): method __init__ (line 109) | def __init__(self, delta=1.0): class TestNormProx (line 113) | class TestNormProx: method test_prox (line 130) | def test_prox(self, norm, alpha, test_prox_obj): method test_conj_prox (line 137) | def test_conj_prox(self, norm, alpha, test_prox_obj): method test_prox_blockarray (line 146) | def test_prox_blockarray(self, norm, alpha, test_prox_obj): method test_prox_zeros (line 159) | def test_prox_zeros(self, norm, test_prox_obj): method test_scaled_attrs (line 166) | def test_scaled_attrs(self, norm, test_prox_obj): method test_scaled_eval (line 176) | def test_scaled_eval(self, norm, alpha, test_prox_obj): method test_scaled_prox (line 185) | def test_scaled_prox(self, norm, alpha, test_prox_obj): class TestBlockArrayEval (line 194) | class TestBlockArrayEval: method test_eval (line 215) | def test_eval(self, cls, test_prox_obj): function test_proj_obj (line 234) | def test_proj_obj(request): class TestProj (line 238) | class TestProj: method test_prox (line 243) | def test_prox(self, cnstr, test_proj_obj): method test_prox_scale_invariance (line 257) | def test_prox_scale_invariance(self, cnstr, test_proj_obj): method test_setdistance (line 267) | def test_setdistance(self, sdist, cnstr, alpha, test_proj_obj): FILE: scico/test/functional/test_indicator.py function test_indicator (line 15) | def test_indicator(indicator): FILE: scico/test/functional/test_loss.py class TestLoss (line 18) | class TestLoss: method setup_method (line 19) | def setup_method(self): method test_generic_squared_l2 (line 36) | def test_generic_squared_l2(self): method test_generic_exception (line 46) | def test_generic_exception(self): method test_squared_l2 (line 57) | def test_squared_l2(self): method test_squared_l2_grad (line 87) | def test_squared_l2_grad(self): method test_weighted_squared_l2 (line 97) | def test_weighted_squared_l2(self): method test_poisson (line 115) | def test_poisson(self): class TestAbsLoss (line 131) | class TestAbsLoss: method setup_method (line 137) | def setup_method(self): method test_properties (line 152) | def test_properties(self, loss_tuple): method test_prox (line 191) | def test_prox(self, loss_tuple): function test_cubic_root (line 221) | def test_cubic_root(): FILE: scico/test/functional/test_misc.py class TestCheckAttrs (line 11) | class TestCheckAttrs: method test_has_eval (line 32) | def test_has_eval(self, cls): method test_has_prox (line 36) | def test_has_prox(self, cls): class TestJit (line 40) | class TestJit: method test_jit (line 62) | def test_jit(self, cls): function test_functional_sum (line 81) | def test_functional_sum(): function test_scalar_vmap (line 91) | def test_scalar_vmap(): function test_scalar_pmap (line 105) | def test_scalar_pmap(): function test_scalar_aggregation (line 118) | def test_scalar_aggregation(): function test_repr_str (line 139) | def test_repr_str(func): FILE: scico/test/functional/test_norm.py function test_l21norm (line 10) | def test_l21norm(axis): function test_l2norm_blockarray (line 31) | def test_l2norm_blockarray(): FILE: scico/test/functional/test_proxavg.py function test_proxavg_init (line 11) | def test_proxavg_init(): function test_proxavg (line 31) | def test_proxavg(): FILE: scico/test/functional/test_separable.py class SeparableTestObject (line 17) | class SeparableTestObject: method __init__ (line 18) | def __init__(self, dtype): function test_separable_obj (line 33) | def test_separable_obj(request): function test_separable_eval (line 37) | def test_separable_eval(test_separable_obj): function test_separable_prox (line 44) | def test_separable_prox(test_separable_obj): function test_separable_grad (line 53) | def test_separable_grad(test_separable_obj): FILE: scico/test/functional/test_tvnorm.py function test_single_axis_haar_transform (line 14) | def test_single_axis_haar_transform(axis): function test_haar_transform (line 20) | def test_haar_transform(): function test_aniso_1d (line 27) | def test_aniso_1d(circular): class Test2D (line 63) | class Test2D: method setup_method (line 64) | def setup_method(self): method test_2d (line 79) | def test_2d(self, tvtype, circular): class Test3D (line 122) | class Test3D: method setup_method (line 123) | def setup_method(self): method test_3d (line 139) | def test_3d(self, tvtype, circular): FILE: scico/test/linop/test_binop.py class TestBinaryOp (line 10) | class TestBinaryOp: method setup_method (line 11) | def setup_method(self, method): method test_case1 (line 16) | def test_case1(self, operator): method test_case2 (line 28) | def test_case2(self, operator): method test_case3 (line 40) | def test_case3(self, operator): method test_case4 (line 52) | def test_case4(self, operator): FILE: scico/test/linop/test_circconv.py class TestCircularConvolve (line 28) | class TestCircularConvolve: method setup_method (line 29) | def setup_method(self, method): method test_eval (line 35) | def test_eval(self, axes_shape_spec, input_dtype, jit): method test_adjoint (line 63) | def test_adjoint(self, axes_shape_spec, input_dtype, jit): method test_scalar_left (line 75) | def test_scalar_left(self, axes_shape_spec, operator, jit): method test_scalar_right (line 92) | def test_scalar_right(self, axes_shape_spec, operator, jit): method test_add_sub (line 107) | def test_add_sub(self, axes_shape_spec, jit): method test_matches_convolve (line 123) | def test_matches_convolve(self, input_dtype, jit): method test_center (line 151) | def test_center(self, center, jit): method test_fractional_center (line 163) | def test_fractional_center(self, jit): method test_from_operator (line 182) | def test_from_operator(self, axes_shape_spec, input_dtype, jit_old_op,... method test_from_operator_block_array (line 194) | def test_from_operator_block_array(self): FILE: scico/test/linop/test_conversions.py function testCircularConvolve_from_FiniteDifference (line 24) | def testCircularConvolve_from_FiniteDifference(shape_axes, input_dtype, ... FILE: scico/test/linop/test_convolve.py class TestConvolve (line 15) | class TestConvolve: method setup_method (line 16) | def setup_method(self, method): method test_eval (line 23) | def test_eval(self, input_shape, input_dtype, mode, jit): method test_adjoint (line 39) | def test_adjoint(self, input_shape, mode, jit, input_dtype): class ConvolveTestObj (line 50) | class ConvolveTestObj: method __init__ (line 51) | def __init__(self): function testobj (line 75) | def testobj(request): function test_init (line 79) | def test_init(testobj): function test_scalar_left (line 89) | def test_scalar_left(testobj, operator): function test_scalar_right (line 97) | def test_scalar_right(testobj, operator): function test_convolve_add_sub (line 107) | def test_convolve_add_sub(testobj, operator): function test_add_sub_different_mode (line 125) | def test_add_sub_different_mode(testobj, operator): function test_add_sum_generic_linop (line 134) | def test_add_sum_generic_linop(testobj, operator): function test_add_sum_conv (line 146) | def test_add_sum_conv(testobj, operator): function test_mul_div_generic_linop (line 158) | def test_mul_div_generic_linop(testobj, operator): function test_invalid_mode (line 164) | def test_invalid_mode(testobj): function test_dimension_mismatch (line 170) | def test_dimension_mismatch(testobj): class TestConvolveByX (line 176) | class TestConvolveByX: method setup_method (line 177) | def setup_method(self, method): method test_eval (line 184) | def test_eval(self, input_shape, input_dtype, mode, jit): method test_adjoint (line 201) | def test_adjoint(self, input_shape, mode, jit, input_dtype): class ConvolveByXTestObj (line 212) | class ConvolveByXTestObj: method __init__ (line 213) | def __init__(self): function cbx_testobj (line 237) | def cbx_testobj(request): function test_cbx_scalar_left (line 242) | def test_cbx_scalar_left(cbx_testobj, operator): function test_cbx_scalar_right (line 250) | def test_cbx_scalar_right(cbx_testobj, operator): function test_convolve_add_sub (line 260) | def test_convolve_add_sub(cbx_testobj, operator): function test_add_sub_different_mode (line 278) | def test_add_sub_different_mode(cbx_testobj, operator): function test_add_sum_generic_linop (line 287) | def test_add_sum_generic_linop(cbx_testobj, operator): function test_add_sum_conv (line 299) | def test_add_sum_conv(cbx_testobj, operator): function test_mul_div_generic_linop (line 311) | def test_mul_div_generic_linop(cbx_testobj, operator): function test_invalid_mode (line 317) | def test_invalid_mode(cbx_testobj): function test_dimension_mismatch (line 323) | def test_dimension_mismatch(cbx_testobj): FILE: scico/test/linop/test_dft.py class TestDFT (line 13) | class TestDFT: method setup_method (line 14) | def setup_method(self, method): method test_dft (line 31) | def test_dft(self, input_shape, axes_and_shape, norm, jit): method test_axes_check (line 60) | def test_axes_check(self): FILE: scico/test/linop/test_diag.py class TestDiagonal (line 21) | class TestDiagonal: method setup_method (line 22) | def setup_method(self, method): method test_eval (line 29) | def test_eval(self, input_shape, diagonal_dtype): method test_eval_broadcasting (line 38) | def test_eval_broadcasting(self, diagonal_dtype): method test_adjoint (line 67) | def test_adjoint(self, input_shape, diagonal_dtype): method test_binary_op (line 77) | def test_binary_op(self, input_shape1, input_shape2, diagonal_dtype, o... method test_matmul (line 97) | def test_matmul(self, input_shape1, input_shape2, diagonal_dtype): method test_binary_op_mismatch (line 117) | def test_binary_op_mismatch(self, operator): method test_scalar_right (line 130) | def test_scalar_right(self, operator): method test_scalar_left (line 147) | def test_scalar_left(self, operator): method test_gram_op (line 161) | def test_gram_op(self, diagonal_dtype): method test_norm (line 173) | def test_norm(self, diagonal_dtype, ord): method test_norm_except (line 183) | def test_norm_except(self): class TestScaledIdentity (line 192) | class TestScaledIdentity: method setup_method (line 193) | def setup_method(self, method): method test_eval (line 200) | def test_eval(self, input_shape, input_dtype): method test_binary_op (line 210) | def test_binary_op(self, input_shape, operator): method test_scale (line 227) | def test_scale(self): method test_norm (line 250) | def test_norm(self, input_dtype, ord): method test_norm_except (line 264) | def test_norm_except(self): class TestIdentity (line 272) | class TestIdentity: method setup_method (line 273) | def setup_method(self, method): method test_eval (line 280) | def test_eval(self, input_shape, input_dtype): method test_binary_op (line 289) | def test_binary_op(self, input_shape, operator): method test_scale (line 315) | def test_scale(self): FILE: scico/test/linop/test_diff.py function test_eval (line 11) | def test_eval(): function test_except (line 38) | def test_except(): function test_eval_prepend (line 52) | def test_eval_prepend(): function test_eval_append (line 60) | def test_eval_append(): function test_adjoint (line 72) | def test_adjoint(input_shape, input_dtype, axes, jit): function test_eval_circular (line 92) | def test_eval_circular(shape_axes, input_dtype, jit): FILE: scico/test/linop/test_func.py function test_transpose (line 11) | def test_transpose(): function test_transpose_ext_init (line 22) | def test_transpose_ext_init(): function test_reshape (line 32) | def test_reshape(): function test_pad (line 43) | def test_pad(): function test_crop (line 59) | def test_crop(): function test_crop_pad_adjoint (line 70) | def test_crop_pad_adjoint(pad): class SliceTestObj (line 77) | class SliceTestObj: method __init__ (line 78) | def __init__(self, dtype): function slicetestobj (line 83) | def slicetestobj(request): function test_slice_eval (line 100) | def test_slice_eval(slicetestobj, idx): function test_slice_adj (line 107) | def test_slice_adj(slicetestobj, idx): function test_slice_blockarray (line 121) | def test_slice_blockarray(idx): FILE: scico/test/linop/test_grad.py function test_proj_grad (line 20) | def test_proj_grad(): class TestPolarGradient (line 40) | class TestPolarGradient: method setup_method (line 41) | def setup_method(self, method): method test_eval (line 59) | def test_eval(self, cdiff, shape_axes, center, outflags, input_dtype, ... class TestCylindricalGradient (line 98) | class TestCylindricalGradient: method setup_method (line 99) | def setup_method(self, method): method test_eval (line 128) | def test_eval(self, shape_axes, center, outflags, input_dtype, jit): class TestSphericalGradient (line 169) | class TestSphericalGradient: method setup_method (line 170) | def setup_method(self, method): method test_eval (line 199) | def test_eval(self, shape_axes, center, outflags, input_dtype, jit): FILE: scico/test/linop/test_linop.py function adjoint_test (line 23) | def adjoint_test( class AbsMatOp (line 41) | class AbsMatOp(linop.LinearOperator): method __init__ (line 49) | def __init__(self, A, adj_fn=None): method _eval (line 55) | def _eval(self, x): class LinearOperatorTestObj (line 59) | class LinearOperatorTestObj: method __init__ (line 60) | def __init__(self, dtype): function testobj (line 80) | def testobj(request): function test_binary_op (line 85) | def test_binary_op(testobj, operator): function test_scalar_left (line 106) | def test_scalar_left(testobj, operator, scalar): function test_scalar_right (line 117) | def test_scalar_right(testobj, operator, scalar): function test_negation (line 126) | def test_negation(testobj): function test_invalid_add_sub_array (line 134) | def test_invalid_add_sub_array(testobj, operator): function test_invalid_add_sub_scalar (line 141) | def test_invalid_add_sub_scalar(testobj, operator): function test_matmul_left (line 147) | def test_matmul_left(testobj): function test_matmul_right (line 154) | def test_matmul_right(testobj): function test_matvec_left (line 161) | def test_matvec_left(testobj): function test_matvec_right (line 168) | def test_matvec_right(testobj): function test_gram (line 175) | def test_gram(testobj): function test_matvec_call (line 188) | def test_matvec_call(testobj): function test_adj_composition (line 193) | def test_adj_composition(testobj): function test_transpose_matvec (line 209) | def test_transpose_matvec(testobj): function test_transpose_matmul (line 224) | def test_transpose_matmul(testobj): function test_conj_transpose_matmul (line 234) | def test_conj_transpose_matmul(testobj): function test_conj_matvec (line 244) | def test_conj_matvec(testobj): function test_adjoint_matvec (line 253) | def test_adjoint_matvec(testobj): function test_adjoint_matmul (line 271) | def test_adjoint_matmul(testobj): function test_hermitian (line 280) | def test_hermitian(testobj): function test_shape (line 287) | def test_shape(testobj): function test_adj_lazy (line 305) | def test_adj_lazy(): function test_jit_adj_lazy (line 318) | def test_jit_adj_lazy(): FILE: scico/test/linop/test_linop_stack.py class TestVerticalStack (line 21) | class TestVerticalStack: method setup_method (line 22) | def setup_method(self, method): method test_construct (line 26) | def test_construct(self, jit): method test_adjoint (line 77) | def test_adjoint(self, collapse_output, jit): method test_algebra (line 92) | def test_algebra(self, collapse_output, jit): class TestBlockDiagonalLinearOperator (line 112) | class TestBlockDiagonalLinearOperator: method test_construct (line 113) | def test_construct(self): method test_apply (line 119) | def test_apply(self): method test_adjoint (line 134) | def test_adjoint(self): method test_input_collapse (line 155) | def test_input_collapse(self): method test_output_collapse (line 166) | def test_output_collapse(self): class TestDiagonalReplicated (line 179) | class TestDiagonalReplicated: method setup_method (line 180) | def setup_method(self, method): method test_adjoint (line 183) | def test_adjoint(self): FILE: scico/test/linop/test_linop_util.py function test_valid_adjoint (line 19) | def test_valid_adjoint(): class PowerIterTestObj (line 31) | class PowerIterTestObj: method __init__ (line 32) | def __init__(self, dtype): function pitestobj (line 50) | def pitestobj(request): function test_power_iteration (line 54) | def test_power_iteration(pitestobj): function test_operator_norm (line 67) | def test_operator_norm(): function test_jacobian (line 84) | def test_jacobian(dtype, inc_eval): FILE: scico/test/linop/test_matrix.py class TestMatrix (line 17) | class TestMatrix: method setup_method (line 18) | def setup_method(self, method): method test_eval (line 24) | def test_eval(self, matrix_shape, input_dtype, input_cols): method test_adjoint (line 39) | def test_adjoint(self, matrix_shape, input_dtype, input_cols): method test_adjoint_method (line 49) | def test_adjoint_method(self, matrix_shape, input_dtype, input_cols): method test_hermetian_method (line 58) | def test_hermetian_method(self, matrix_shape, input_dtype, input_cols): method test_gram_method (line 67) | def test_gram_method(self, matrix_shape, input_dtype, input_cols): method test_gram_op (line 76) | def test_gram_op(self, matrix_shape, input_dtype, input_cols): method test_add_sub (line 84) | def test_add_sub(self, operator): method test_scalar_left (line 101) | def test_scalar_left(self, operator): method test_scalar_right (line 111) | def test_scalar_right(self, operator): method test_elementwise_matops (line 121) | def test_elementwise_matops(self, operator): method test_elementwise_array_left (line 131) | def test_elementwise_array_left(self, operator): method test_elementwise_array_right (line 139) | def test_elementwise_array_right(self, operator): method test_elementwise_matop_shape_mismatch (line 147) | def test_elementwise_matop_shape_mismatch(self, operator): method test_elementwise_array_shape_mismatch (line 156) | def test_elementwise_array_shape_mismatch(self, operator): method test_elementwise_linop (line 168) | def test_elementwise_linop(self, operator): method test_elementwise_linop_mismatch (line 178) | def test_elementwise_linop_mismatch(self, operator): method test_elementwise_linop_invalid (line 187) | def test_elementwise_linop_invalid(self, operator): method test_matmul (line 198) | def test_matmul(self): method test_matmul_cols (line 208) | def test_matmul_cols(self): method test_matmul_linop (line 218) | def test_matmul_linop(self): method test_matmul_linop_shape_mismatch (line 228) | def test_matmul_linop_shape_mismatch(self): method test_matmul_identity (line 236) | def test_matmul_identity(self): method test_init_array (line 242) | def test_init_array(self): method test_init_wrong_dims (line 255) | def test_init_wrong_dims(self, matrix_shape): method test_to_array (line 260) | def test_to_array(self): method test_norm (line 275) | def test_norm(self, ord, axis, keepdims, input_dtype): # pylint: disa... FILE: scico/test/linop/test_optics.py class TestPropagator (line 19) | class TestPropagator: method setup_method (line 20) | def setup_method(self, method): method test_prop_adjoint (line 30) | def test_prop_adjoint(self, prop, ndim): method test_AS_inverse (line 35) | def test_AS_inverse(self, ndim): method test_3d_invalid (line 45) | def test_3d_invalid(self, prop): method test_shape_dx_mismatch (line 50) | def test_shape_dx_mismatch(self, prop): method test_3d_invalid_radial (line 54) | def test_3d_invalid_radial(self): method test_shape_dx_mismatch_radial (line 58) | def test_shape_dx_mismatch_radial(self): function test_asp_sampling (line 64) | def test_asp_sampling(ndim): function test_fresnel_sampling (line 75) | def test_fresnel_sampling(ndim): function test_fraunhofer_sampling (line 86) | def test_fraunhofer_sampling(ndim): FILE: scico/test/linop/xray/test_abel.py function make_im (line 15) | def make_im(Nx, Ny): function test_inverse (line 24) | def test_inverse(Nx, Ny): function test_adjoint (line 35) | def test_adjoint(Nx, Ny): function test_ATA (line 42) | def test_ATA(Nx, Ny): function test_grad (line 51) | def test_grad(Nx, Ny): function test_adjoint_grad (line 61) | def test_adjoint_grad(Nx, Ny): FILE: scico/test/linop/xray/test_astra.py function make_im (line 34) | def make_im(Nx, Ny, is_3d=True): function get_tol (line 45) | def get_tol(): function get_tol_random_input (line 53) | def get_tol_random_input(): class XRayTransform2DTest (line 61) | class XRayTransform2DTest: method __init__ (line 62) | def __init__(self, volume_geometry): function testobj (line 81) | def testobj(request): function test_init (line 85) | def test_init(testobj): function test_ATA_call (line 110) | def test_ATA_call(testobj): function test_ATA_matmul (line 117) | def test_ATA_matmul(testobj): function test_AAT_call (line 124) | def test_AAT_call(testobj): function test_AAT_matmul (line 131) | def test_AAT_matmul(testobj): function test_grad (line 138) | def test_grad(testobj): function test_adjoint_grad (line 149) | def test_adjoint_grad(testobj): function test_adjoint_random (line 157) | def test_adjoint_random(testobj): function test_adjoint_typical_input (line 162) | def test_adjoint_typical_input(testobj): function test_fbp (line 169) | def test_fbp(testobj): function test_jit_in_DiagonalStack (line 176) | def test_jit_in_DiagonalStack(): function test_3D_on_GPU (line 184) | def test_3D_on_GPU(): function test_3D_api_equiv (line 196) | def test_3D_api_equiv(): function test_angle_to_vector (line 209) | def test_angle_to_vector(): function test_rotate_vectors (line 216) | def test_rotate_vectors(): function test_geometry (line 226) | def test_geometry(): function test_projection_convention (line 263) | def test_projection_convention(test_geometry): function test_project_coords (line 286) | def test_project_coords(test_geometry): function test_convert_to_scico_geometry (line 303) | def test_convert_to_scico_geometry(test_geometry): function test_convert_from_scico_geometry (line 313) | def test_convert_from_scico_geometry(test_geometry): function test_vol_coord_to_world_coord (line 328) | def test_vol_coord_to_world_coord(): function test_ensure_writeable (line 335) | def test_ensure_writeable(): FILE: scico/test/linop/xray/test_svmbir.py function pytest_generate_tests (line 30) | def pytest_generate_tests(metafunc): function make_im (line 62) | def make_im(Nx, Ny, is_3d=True): function make_angles (line 73) | def make_angles(num_angles): function make_A (line 77) | def make_A( function test_grad (line 104) | def test_grad( function test_adjoint (line 132) | def test_adjoint( function test_prox (line 155) | def test_prox( function test_prox_weights (line 186) | def test_prox_weights( function test_prox_cg (line 220) | def test_prox_cg( function test_approx_prox (line 270) | def test_approx_prox( FILE: scico/test/linop/xray/test_symcone.py class TestAxialSymm (line 15) | class TestAxialSymm: method setup_method (line 16) | def setup_method(self, method): method test_vbas (line 25) | def test_vbas(self, axis): class TestAbelCone (line 44) | class TestAbelCone: method setup_method (line 45) | def setup_method(self, method): method test_2d (line 54) | def test_2d(self, num_slabs): method test_2d_unequal (line 62) | def test_2d_unequal(self, num_slabs): method test_3d (line 71) | def test_3d(self, num_slabs): method test_3d_unequal (line 78) | def test_3d_unequal(self, num_slabs): method test_2d3d_unequal (line 86) | def test_2d3d_unequal(self, num_slabs): method test_proj_axis (line 94) | def test_proj_axis(self, axis): method test_fdk (line 113) | def test_fdk(self, axis): FILE: scico/test/linop/xray/test_xray_2d.py function test_init (line 15) | def test_init(): function test_apply (line 33) | def test_apply(): function test_apply_adjoint (line 52) | def test_apply_adjoint(): function test_matched_adjoint (line 77) | def test_matched_adjoint(): function test_fbp (line 90) | def test_fbp(dx, det_count_factor): function test_fbp_jit (line 105) | def test_fbp_jit(): FILE: scico/test/linop/xray/test_xray_3d.py function test_matched_adjoint (line 9) | def test_matched_adjoint(): function test_scaling (line 29) | def test_scaling(): FILE: scico/test/linop/xray/test_xray_util.py function test_image_centroid (line 29) | def test_image_centroid(): function test_center_image (line 36) | def test_center_image(): function test_rotate_volume (line 45) | def test_rotate_volume(): function align_test_tol (line 52) | def align_test_tol(): function test_image_alignment (line 61) | def test_image_alignment(): function test_volume_alignment (line 73) | def test_volume_alignment(): FILE: scico/test/numpy/test_blockarray.py function make_arbitrary_jax_array (line 22) | def make_arbitrary_jax_array(shape, dtype): function sequence_assert_allclose (line 29) | def sequence_assert_allclose(x, y, *args, **kwargs): class OperatorsTestObj (line 37) | class OperatorsTestObj: method __init__ (line 40) | def __init__(self, dtype): function test_operator_obj (line 72) | def test_operator_obj(request): function test_operator_left (line 78) | def test_operator_left(test_operator_obj, operator): function test_operator_right (line 87) | def test_operator_right(test_operator_obj, operator): function test_ba_ba_operator (line 97) | def test_ba_ba_operator(test_operator_obj, operator): function test_ba_ba_matmul (line 107) | def test_ba_ba_matmul(test_operator_obj): function test_conj (line 127) | def test_conj(test_operator_obj): function test_real (line 135) | def test_real(test_operator_obj): function test_imag (line 142) | def test_imag(test_operator_obj): function test_ndim (line 149) | def test_ndim(test_operator_obj): function test_getitem (line 154) | def test_getitem(test_operator_obj): function test_split (line 175) | def test_split(test_operator_obj): function test_blockarray_from_one_array (line 181) | def test_blockarray_from_one_array(): function test_sum_method (line 190) | def test_sum_method(test_operator_obj, axis, keepdims): function test_eval_shape_1arg (line 199) | def test_eval_shape_1arg(test_operator_obj): function test_eval_shape_2arg (line 210) | def test_eval_shape_2arg(test_operator_obj): function test_linear_transpose (line 227) | def test_linear_transpose(test_operator_obj): function test_ba_ba_dot (line 236) | def test_ba_ba_dot(test_operator_obj, operator): class BlockArrayReductionObj (line 258) | class BlockArrayReductionObj: method __init__ (line 259) | def __init__(self, dtype): function reduction_obj (line 275) | def reduction_obj(request): function test_reduce (line 291) | def test_reduce(reduction_obj, func): function test_reduce_axis (line 301) | def test_reduce_axis(reduction_obj, func, axis): function test_reduce_singleton (line 316) | def test_reduce_singleton(reduction_obj, func): class TestCreators (line 330) | class TestCreators: method setup_method (line 331) | def setup_method(self, method): method test_zeros (line 339) | def test_zeros(self): method test_empty (line 344) | def test_empty(self): method test_ones (line 349) | def test_ones(self): method test_full (line 354) | def test_full(self): method test_full_nodtype (line 361) | def test_full_nodtype(self): function test_list_triggering (line 369) | def test_list_triggering(): function test_test_func (line 378) | def test_test_func(func): function x (line 388) | def x(): function y (line 394) | def y(): function test_unary (line 400) | def test_unary(op, x): function test_elementwise_binary (line 424) | def test_elementwise_binary(op, x, y): function test_not_implemented_binary (line 431) | def test_not_implemented_binary(x): function test_matmul (line 436) | def test_matmul(x): function test_property (line 446) | def test_property(): function test_method (line 453) | def test_method(): function test_stack (line 461) | def test_stack(): function test_ravel (line 470) | def test_ravel(): FILE: scico/test/numpy/test_numpy.py function on_cpu (line 11) | def on_cpu(): function check_results (line 15) | def check_results(jout, sout): function test_reshape_array (line 31) | def test_reshape_array(): function test_ufunc_abs (line 36) | def test_ufunc_abs(): function test_ufunc_maximum (line 58) | def test_ufunc_maximum(): function test_ufunc_sign (line 102) | def test_ufunc_sign(): function test_ufunc_where (line 116) | def test_ufunc_where(): function test_ufunc_true_divide (line 140) | def test_ufunc_true_divide(): function test_ufunc_floor_divide (line 167) | def test_ufunc_floor_divide(): function test_ufunc_real (line 194) | def test_ufunc_real(): function test_ufunc_imag (line 212) | def test_ufunc_imag(): function test_ufunc_conj (line 230) | def test_ufunc_conj(): function test_create_zeros (line 248) | def test_create_zeros(): function test_create_ones (line 264) | def test_create_ones(): function test_create_empty (line 272) | def test_create_empty(): function test_create_full (line 280) | def test_create_full(): function test_create_zeros_like (line 291) | def test_create_zeros_like(): function test_create_empty_like (line 307) | def test_create_empty_like(): function test_create_ones_like (line 321) | def test_create_ones_like(): function test_create_full_like (line 335) | def test_create_full_like(): function test_wrap_recursively (line 349) | def test_wrap_recursively(): function test_add_full_reduction (line 357) | def test_add_full_reduction(): FILE: scico/test/numpy/test_numpy_util.py function test_ntpl_list_transpose (line 35) | def test_ntpl_list_transpose(): function test_namedtuple_to_array (line 44) | def test_namedtuple_to_array(): function test_no_nan_divide_array (line 52) | def test_no_nan_divide_array(): function test_no_nan_divide_blockarray (line 64) | def test_no_nan_divide_blockarray(): function test_array_info (line 76) | def test_array_info(): function test_normalize_axes (line 88) | def test_normalize_axes(): function test_slice_length (line 124) | def test_slice_length(length, start, stop, stride): function test_slice_length_other (line 132) | def test_slice_length_other(length, slc): function test_indexed_shape (line 157) | def test_indexed_shape(shape, slc): function test_is_nested (line 163) | def test_is_nested(): function test_is_collapsible (line 186) | def test_is_collapsible(): function test_is_blockable (line 193) | def test_is_blockable(): function test_shape_dtype_rep (line 201) | def test_shape_dtype_rep(shape): function test_is_real_dtype (line 206) | def test_is_real_dtype(): function test_is_complex_dtype (line 211) | def test_is_complex_dtype(): function test_real_dtype (line 216) | def test_real_dtype(): function test_complex_dtype (line 220) | def test_complex_dtype(): function test_dtype_name (line 224) | def test_dtype_name(): function test_broadcast_nested_shapes (line 229) | def test_broadcast_nested_shapes(): function test_is_scalar_equiv (line 246) | def test_is_scalar_equiv(): FILE: scico/test/operator/test_biconvolve.py class TestBiConvolve (line 14) | class TestBiConvolve: method setup_method (line 15) | def setup_method(self, method): method test_eval (line 21) | def test_eval(self, input_dtype, mode, jit): method test_invalid_shapes (line 43) | def test_invalid_shapes(self): FILE: scico/test/operator/test_op_stack.py class TestVerticalStack (line 26) | class TestVerticalStack: method setup_method (line 27) | def setup_method(self, method): method test_construct (line 31) | def test_construct(self, jit): method test_algebra (line 77) | def test_algebra(self, collapse_output, jit): class TestBlockDiagonalOperator (line 97) | class TestBlockDiagonalOperator: method test_construct (line 98) | def test_construct(self): method test_apply (line 115) | def test_apply(self): method test_input_collapse (line 130) | def test_input_collapse(self): method test_output_collapse (line 141) | def test_output_collapse(self): class TestDiagonalReplicated (line 152) | class TestDiagonalReplicated: method setup_method (line 153) | def setup_method(self, method): method test_map_auto_vmap (line 158) | def test_map_auto_vmap(self, input_axis, map_type): method test_map_auto_pmap (line 168) | def test_map_auto_pmap(self): method test_input_axis (line 176) | def test_input_axis(self): method test_output_axis (line 188) | def test_output_axis(self): FILE: scico/test/operator/test_operator.py class AbsOperator (line 21) | class AbsOperator(Operator): method _eval (line 22) | def _eval(self, x): class SquareOperator (line 26) | class SquareOperator(Operator): method _eval (line 27) | def _eval(self, x): class SumSquareOperator (line 31) | class SumSquareOperator(Operator): method _eval (line 32) | def _eval(self, x): class OperatorTestObj (line 36) | class OperatorTestObj: method __init__ (line 37) | def __init__(self, dtype): function testobj (line 53) | def testobj(request): function test_binary_op (line 58) | def test_binary_op(testobj, operator): function test_binary_op_same (line 74) | def test_binary_op_same(testobj, operator): function test_scalar_left (line 89) | def test_scalar_left(testobj, operator, scalar): function test_scalar_right (line 99) | def test_scalar_right(testobj, operator, scalar): function test_negation (line 109) | def test_negation(testobj): function test_invalid_add_sub_array (line 118) | def test_invalid_add_sub_array(testobj, operator): function test_invalid_add_sub_scalar (line 125) | def test_invalid_add_sub_scalar(testobj, operator): function test_call_operator_operator (line 131) | def test_call_operator_operator(testobj): function test_shape_call_vec (line 142) | def test_shape_call_vec(testobj): function test_scale_vmap (line 148) | def test_scale_vmap(testobj): function test_scale_pmap (line 161) | def test_scale_pmap(testobj): function test_freeze_3arg (line 174) | def test_freeze_3arg(): function test_freeze_2arg (line 200) | def test_freeze_2arg(): function test_func_op (line 219) | def test_func_op(op_fn, dtype): function test_make_func_op (line 228) | def test_make_func_op(): function test_make_func_op_ext_init (line 236) | def test_make_func_op_ext_init(): class TestJacobianProdReal (line 246) | class TestJacobianProdReal: method setup_method (line 247) | def setup_method(self): method test_jvp (line 264) | def test_jvp(self): method test_vjp_conj (line 269) | def test_vjp_conj(self): method test_vjp_noconj (line 275) | def test_vjp_noconj(self): class TestJacobianProdComplex (line 282) | class TestJacobianProdComplex: method setup_method (line 283) | def setup_method(self): method test_jvp (line 300) | def test_jvp(self): method test_vjp_conj (line 305) | def test_vjp_conj(self): method test_vjp_noconj (line 311) | def test_vjp_noconj(self): FILE: scico/test/optimize/test_admm.py class TestMisc (line 21) | class TestMisc: method setup_method (line 22) | def setup_method(self, method): method test_admm (line 26) | def test_admm(self): method test_admm_aux (line 80) | def test_admm_aux(self, solver): class TestReal (line 109) | class TestReal: method setup_method (line 110) | def setup_method(self, method): method test_admm_generic (line 130) | def test_admm_generic(self): method test_admm_saveload (line 151) | def test_admm_saveload(self): method test_admm_quadratic_scico (line 194) | def test_admm_quadratic_scico(self): method test_admm_quadratic_jax (line 215) | def test_admm_quadratic_jax(self): method test_admm_quadratic_relax (line 236) | def test_admm_quadratic_relax(self): class TestRealWeighted (line 259) | class TestRealWeighted: method setup_method (line 260) | def setup_method(self, method): method test_admm_quadratic_linear (line 283) | def test_admm_quadratic_linear(self): method test_admm_quadratic_matrix (line 304) | def test_admm_quadratic_matrix(self): class TestComplex (line 326) | class TestComplex: method setup_method (line 327) | def setup_method(self, method): method test_admm_generic (line 346) | def test_admm_generic(self): method test_admm_quadratic_linear (line 367) | def test_admm_quadratic_linear(self): method test_admm_quadratic_matrix (line 390) | def test_admm_quadratic_matrix(self): class TestCircularConvolveSolve (line 414) | class TestCircularConvolveSolve: method setup_and_teardown (line 417) | def setup_and_teardown(self, extra_axis, center): method test_admm (line 436) | def test_admm(self): class TestSpecialCaseCircularConvolveSolve (line 468) | class TestSpecialCaseCircularConvolveSolve: method setup_and_teardown (line 471) | def setup_and_teardown(self, with_cconv): method test_admm (line 488) | def test_admm(self): class TestBlockCircularConvolveSolve (line 520) | class TestBlockCircularConvolveSolve: method setup_method (line 521) | def setup_method(self, method): method test_fblock_init (line 540) | def test_fblock_init(self): method test_g0block_init (line 569) | def test_g0block_init(self): method test_solve (line 598) | def test_solve(self): FILE: scico/test/optimize/test_ladmm.py class TestMisc (line 14) | class TestMisc: method setup_method (line 15) | def setup_method(self, method): method test_itstat (line 26) | def test_itstat(self): method test_callback (line 54) | def test_callback(self): method test_finite_check (line 71) | def test_finite_check(self): class TestBlockArray (line 81) | class TestBlockArray: method setup_method (line 82) | def setup_method(self, method): method test_blockarray (line 101) | def test_blockarray(self): class TestReal (line 114) | class TestReal: method setup_method (line 115) | def setup_method(self, method): method test_ladmm (line 132) | def test_ladmm(self): method test_ladmm_saveload (line 152) | def test_ladmm_saveload(self): class TestComplex (line 191) | class TestComplex: method setup_method (line 192) | def setup_method(self, method): method test_ladmm (line 209) | def test_ladmm(self): FILE: scico/test/optimize/test_padmm.py class TestMisc (line 14) | class TestMisc: method setup_method (line 15) | def setup_method(self, method): method test_itstat_padmm (line 34) | def test_itstat_padmm(self): method test_itstat_nlpadmm (line 68) | def test_itstat_nlpadmm(self): method test_callback (line 101) | def test_callback(self): method test_finite_check (line 119) | def test_finite_check(self): class TestBlockArray (line 136) | class TestBlockArray: method setup_method (line 137) | def setup_method(self, method): method test_blockarray_padmm (line 164) | def test_blockarray_padmm(self): method test_blockarray_nlpadmm (line 177) | def test_blockarray_nlpadmm(self): class TestReal (line 191) | class TestReal: method setup_method (line 192) | def setup_method(self, method): method test_padmm (line 209) | def test_padmm(self): method test_padmm_saveload (line 230) | def test_padmm_saveload(self): method test_nlpadmm (line 269) | def test_nlpadmm(self): class TestComplex (line 298) | class TestComplex: method setup_method (line 299) | def setup_method(self, method): method test_nlpadmm (line 316) | def test_nlpadmm(self): class TestEstimateParameters (line 345) | class TestEstimateParameters: method setup_method (line 346) | def setup_method(self): method test_padmm_a (line 364) | def test_padmm_a(self): method test_padmm_ab (line 369) | def test_padmm_ab(self): method test_real (line 374) | def test_real(self): method test_complex (line 379) | def test_complex(self): FILE: scico/test/optimize/test_pdhg.py class TestMisc (line 14) | class TestMisc: method setup_method (line 15) | def setup_method(self, method): method test_itstat (line 26) | def test_itstat(self): method test_callback (line 54) | def test_callback(self): method test_finite_check (line 71) | def test_finite_check(self): class TestBlockArray (line 87) | class TestBlockArray: method setup_method (line 88) | def setup_method(self, method): method test_blockarray (line 107) | def test_blockarray(self): class TestReal (line 120) | class TestReal: method setup_method (line 121) | def setup_method(self, method): method test_pdhg (line 138) | def test_pdhg(self): method test_pdhg_saveload (line 158) | def test_pdhg_saveload(self): method test_nlpdhg (line 196) | def test_nlpdhg(self): class TestComplex (line 219) | class TestComplex: method setup_method (line 220) | def setup_method(self, method): method test_pdhg (line 237) | def test_pdhg(self): class TestEstimateParameters (line 258) | class TestEstimateParameters: method setup_method (line 259) | def setup_method(self): method test_operators_dlft (line 268) | def test_operators_dlft(self): method test_operators (line 274) | def test_operators(self): method test_ratio (line 281) | def test_ratio(self): FILE: scico/test/optimize/test_pgm.py class TestSet (line 21) | class TestSet: method setup_method (line 22) | def setup_method(self, method): method test_pgm (line 38) | def test_pgm(self): method test_pgm_saveload (line 48) | def test_pgm_saveload(self): method test_pgm_isfinite (line 68) | def test_pgm_isfinite(self): method test_accelerated_pgm (line 80) | def test_accelerated_pgm(self): method test_accelerated_pgm_saveload (line 90) | def test_accelerated_pgm_saveload(self): method test_accelerated_pgm_isfinite (line 110) | def test_accelerated_pgm_isfinite(self): method test_pgm_BB_step_size (line 122) | def test_pgm_BB_step_size(self): method test_pgm_adaptive_BB_step_size (line 139) | def test_pgm_adaptive_BB_step_size(self): method test_accelerated_pgm_line_search (line 155) | def test_accelerated_pgm_line_search(self): method test_accelerated_pgm_robust_line_search (line 172) | def test_accelerated_pgm_robust_line_search(self): method test_pgm_BB_step_size_jit (line 189) | def test_pgm_BB_step_size_jit(self): method test_accelerated_pgm_adaptive_BB_step_size_jit (line 211) | def test_accelerated_pgm_adaptive_BB_step_size_jit(self): class TestComplex (line 234) | class TestComplex: method setup_method (line 235) | def setup_method(self, method): method test_pgm (line 251) | def test_pgm(self): method test_accelerated_pgm (line 267) | def test_accelerated_pgm(self): method test_pgm_BB_step_size (line 277) | def test_pgm_BB_step_size(self): method test_pgm_adaptive_BB_step_size (line 294) | def test_pgm_adaptive_BB_step_size(self): method test_accelerated_pgm_line_search (line 311) | def test_accelerated_pgm_line_search(self): method test_accelerated_pgm_robust_line_search (line 328) | def test_accelerated_pgm_robust_line_search(self): FILE: scico/test/osver.py function osx_ver_geq_than (line 6) | def osx_ver_geq_than(verstr): FILE: scico/test/test_core.py class GradTestObj (line 12) | class GradTestObj: method __init__ (line 13) | def __init__(self, dtype): function testobj (line 26) | def testobj(request): function test_grad (line 30) | def test_grad(testobj): function test_grad_aux (line 42) | def test_grad_aux(testobj): function test_value_and_grad (line 57) | def test_value_and_grad(testobj): function test_value_and_grad_aux (line 72) | def test_value_and_grad_aux(testobj): function test_linear_transpose (line 91) | def test_linear_transpose(shape): function test_linear_adjoint_shape (line 105) | def test_linear_adjoint_shape(shape): function test_linear_adjoint (line 118) | def test_linear_adjoint(testobj): function test_linear_adjoint_r_to_c (line 140) | def test_linear_adjoint_r_to_c(): function test_linear_adjoint_c_to_r (line 151) | def test_linear_adjoint_c_to_r(): function test_cvjp (line 164) | def test_cvjp(dtype): function test_eval_shape_1 (line 195) | def test_eval_shape_1(argskwargs): function test_eval_shape_2 (line 227) | def test_eval_shape_2(arrdts): FILE: scico/test/test_data.py class TestSet (line 19) | class TestSet: method test_kodim23_uint (line 20) | def test_kodim23_uint(self): method test_kodim23_float (line 25) | def test_kodim23_float(self): FILE: scico/test/test_denoiser.py function module_setup_teardown (line 16) | def module_setup_teardown(request): class TestBM3D (line 25) | class TestBM3D: method setup_method (line 26) | def setup_method(self): method test_shape (line 31) | def test_shape(self): method test_gry (line 35) | def test_gry(self): method test_rgb (line 43) | def test_rgb(self): method test_bad_inputs (line 51) | def test_bad_inputs(self): class TestBM4D (line 73) | class TestBM4D: method setup_method (line 74) | def setup_method(self): method test_shape (line 80) | def test_shape(self): method test_jit (line 87) | def test_jit(self): method test_bad_inputs (line 102) | def test_bad_inputs(self): class TestDnCNN (line 120) | class TestDnCNN: method setup_method (line 121) | def setup_method(self): method test_single_channel (line 127) | def test_single_channel(self): method test_multi_channel (line 134) | def test_multi_channel(self): method test_init (line 141) | def test_init(self): method test_bad_inputs (line 149) | def test_bad_inputs(self): class TestNonBLindDnCNN (line 164) | class TestNonBLindDnCNN: method setup_method (line 165) | def setup_method(self): method test_single_channel (line 172) | def test_single_channel(self): method test_multi_channel (line 176) | def test_multi_channel(self): method test_bad_inputs (line 180) | def test_bad_inputs(self): FILE: scico/test/test_diagnostics.py class TestSet (line 8) | class TestSet: method test_itstat (line 9) | def test_itstat(self): method test_display (line 18) | def test_display(self, capsys): method test_exception (line 30) | def test_exception(self): method test_warning (line 36) | def test_warning(self): FILE: scico/test/test_examples.py function test_rgb2gray (line 30) | def test_rgb2gray(): function test_volume_read (line 36) | def test_volume_read(): function test_epfl_deconv_data (line 46) | def test_epfl_deconv_data(): function test_ucb_diffusercam_data (line 55) | def test_ucb_diffusercam_data(): function test_downsample_volume (line 64) | def test_downsample_volume(): function test_tile_volume_slices (line 76) | def test_tile_volume_slices(): function test_gaussian (line 85) | def test_gaussian(): function test_create_circular_phantom (line 94) | def test_create_circular_phantom(): function test_create_cone (line 113) | def test_create_cone(img_shape): function test_create_3d_foam_phantom (line 129) | def test_create_3d_foam_phantom(img_shape, N_sphere): function test_conv_sparse_phantom (line 134) | def test_conv_sparse_phantom(): function test_tangle_phantom (line 141) | def test_tangle_phantom(): function test_spnoise (line 146) | def test_spnoise(): function test_phase_diff (line 157) | def test_phase_diff(): FILE: scico/test/test_function.py class TestFunction (line 11) | class TestFunction: method setup_method (line 12) | def setup_method(self): method test_init (line 20) | def test_init(self): method test_eval (line 26) | def test_eval(self): method test_eval_jit (line 36) | def test_eval_jit(self): method test_slice (line 47) | def test_slice(self): method test_join (line 52) | def test_join(self): method test_join_raise (line 57) | def test_join_raise(self): function test_jacobian (line 66) | def test_jacobian(dtype): FILE: scico/test/test_metric.py class TestSet (line 7) | class TestSet: method setup_method (line 8) | def setup_method(self, method): method test_mae_mse (line 11) | def test_mae_mse(self): method test_snr_nrm (line 22) | def test_snr_nrm(self): method test_snr_signal_range (line 29) | def test_snr_signal_range(self): method test_psnr (line 37) | def test_psnr(self): method test_isnr (line 43) | def test_isnr(self): method test_bsnr (line 49) | def test_bsnr(self): function test_rel_res (line 59) | def test_rel_res(): FILE: scico/test/test_random.py function test_wrapped_funcs (line 11) | def test_wrapped_funcs(seed): function test_add_seed_adapter (line 31) | def test_add_seed_adapter(): FILE: scico/test/test_ray_tune.py function test_random_run (line 15) | def test_random_run(): function test_random_tune (line 40) | def test_random_tune(): function test_hyperopt_run (line 65) | def test_hyperopt_run(): function test_hyperopt_tune (line 88) | def test_hyperopt_tune(): function test_hyperopt_tune_alt_init (line 112) | def test_hyperopt_tune_alt_init(): FILE: scico/test/test_scipy_special.py function test_one_arg_funcs (line 29) | def test_one_arg_funcs(func): function test_betainc (line 42) | def test_betainc(): function test_gammainc (line 55) | def test_gammainc(func): function test_multigammaln (line 67) | def test_multigammaln(): function test_logs (line 79) | def test_logs(func): function test_zeta (line 90) | def test_zeta(): FILE: scico/test/test_solver.py class TestSet (line 11) | class TestSet: method setup_method (line 12) | def setup_method(self, method): method test_wrap_func_and_grad (line 15) | def test_wrap_func_and_grad(self): method test_cg_std (line 28) | def test_cg_std(self): method test_cg_op (line 45) | def test_cg_op(self): method test_cg_no_info (line 61) | def test_cg_no_info(self): method test_cg_complex (line 77) | def test_cg_complex(self): method test_preconditioned_cg (line 93) | def test_preconditioned_cg(self): method test_lstsq_func (line 116) | def test_lstsq_func(self): method test_lstsq_op (line 133) | def test_lstsq_op(self): class TestOptimizeScalar (line 149) | class TestOptimizeScalar: method setup_method (line 152) | def setup_method(self): method fun (line 156) | def fun(self, x, a=1.5): method test_minimize_scalar (line 161) | def test_minimize_scalar(self): function test_minimize_vector (line 207) | def test_minimize_vector(dtype, method): function test_minimize_blockarray (line 230) | def test_minimize_blockarray(dtype, method): function test_split_join_array (line 249) | def test_split_join_array(): function test_split_join_blockarray (line 260) | def test_split_join_blockarray(): function test_bisect (line 274) | def test_bisect(): function test_golden (line 289) | def test_golden(): function test_solve_atai (line 305) | def test_solve_atai(cho_factor, wide, weighted, alpha): function test_solve_aati (line 332) | def test_solve_aati(cho_factor, wide, alpha): function test_solve_atad (line 351) | def test_solve_atad(cho_factor, wide, vector): FILE: scico/test/test_util.py function test_rattr (line 22) | def test_rattr(): function test_partial_pos (line 39) | def test_partial_pos(): function test_partial_kw (line 47) | def test_partial_kw(): function test_partial_pos_and_kw (line 55) | def test_partial_pos_and_kw(): function _internet_connected (line 64) | def _internet_connected(host="8.8.8.8", port=53, timeout=3): function test_url_get (line 80) | def test_url_get(): function test_check_for_tracer (line 100) | def test_check_for_tracer(): function test_timer_basic (line 121) | def test_timer_basic(): function test_timer_multi (line 133) | def test_timer_multi(): function test_timer_reset (line 145) | def test_timer_reset(): function test_ctxttimer_basic (line 154) | def test_ctxttimer_basic(): function test_ctxttimer_stopstart (line 161) | def test_ctxttimer_stopstart(): FILE: scico/test/test_version.py function test_var_val (line 7) | def test_var_val(): FILE: scico/trace.py function _get_hash (line 75) | def _get_hash(val: Any) -> Optional[int]: function _trace_arg_repr (line 96) | def _trace_arg_repr(val: Any) -> str: function register_variable (line 141) | def register_variable(var: Any, name: str): function _call_wrapped_function (line 159) | def _call_wrapped_function(func: Callable, *args, **kwargs) -> Any: function call_trace (line 193) | def call_trace(func: Callable) -> Callable: function _submodule_name (line 279) | def _submodule_name(module, obj): function _is_scico_object (line 290) | def _is_scico_object(obj: Any) -> bool: function _is_scico_module (line 303) | def _is_scico_module(mod: types.ModuleType) -> bool: function _in_module (line 315) | def _in_module(mod: types.ModuleType, obj: Any) -> bool: function _is_submodule (line 328) | def _is_submodule(mod: types.ModuleType, submod: types.ModuleType) -> bool: function apply_decorator (line 341) | def apply_decorator( function trace_scico_calls (line 444) | def trace_scico_calls(verbose: bool = False): FILE: scico/util.py function rgetattr (line 23) | def rgetattr(obj: object, name: str, default: Optional[Any] = None) -> Any: function rsetattr (line 46) | def rsetattr(obj: object, name: str, value: Any): function partial (line 61) | def partial(func: Callable, indices: Sequence, *fixargs: Any, **fixkwarg... function device_info (line 109) | def device_info(devid: int = 0) -> str: # pragma: no cover function check_for_tracer (line 129) | def check_for_tracer(func: Callable) -> Callable: function url_get (line 149) | def url_get( class Timer (line 188) | class Timer: method __init__ (line 195) | def __init__( method start (line 227) | def start(self, labels: Optional[Union[str, List[str]]] = None): method stop (line 257) | def stop(self, labels: Optional[Union[str, List[str]]] = None): method reset (line 294) | def reset(self, labels: Optional[Union[str, List[str]]] = None): method elapsed (line 325) | def elapsed(self, label: Optional[str] = None, total: bool = True) -> ... method labels (line 365) | def labels(self) -> List[str]: method __str__ (line 374) | def __str__(self) -> str: class ContextTimer (line 407) | class ContextTimer: method __init__ (line 427) | def __init__( method __enter__ (line 454) | def __enter__(self): method __exit__ (line 463) | def __exit__(self, exc_type, exc_value, traceback): method elapsed (line 474) | def elapsed(self, total: bool = True) -> float: