tfmri.math

Math operations.

Classes

Functions

block_soft_threshold

Block soft thresholding operator.

extract_from_complex

Extract parts from a complex tensor.

indicator_ball

Indicator function of the Lp ball.

indicator_box

Indicator function of a box.

indicator_simplex

Indicator function of the simplex.

make_val_and_grad_fn

Function decorator to compute both function value and gradient.

normalize_no_nan

Normalizes tensor along dimension axis using specified norm.

project_onto_ball

Projects an input vector onto the Lp ball.

project_onto_box

Projects an input vector onto the box.

project_onto_simplex

Projects an input vector onto the simplex.

shrinkage

Shrinkage operator.

soft_threshold

Soft thresholding operator.