Skip to main content
Back to top
Ctrl
+
K
Examples
User Guide
API Reference
About Us
Contributing
Developer Docs
System Settings
Light
Dark
GitHub
Discourse Forum
Choose version
Examples
User Guide
API Reference
About Us
Contributing
Developer Docs
System Settings
Light
Dark
GitHub
Discourse Forum
Choose version
Collapse Sidebar
Expand Sidebar
Section Navigation
adapters
transforms
AsSet
AsTimeSeries
Broadcast
Concatenate
Constrain
ConvertDType
Drop
ElementwiseTransform
ExpandDims
FilterTransform
Group
Keep
Log
MapTransform
NanToNum
NumpyTransform
OneHot
RandomSubsample
Rename
Scale
SerializableCustomTransform
Shift
Split
Sqrt
Squeeze
Standardize
Take
ToArray
ToDict
Transform
Ungroup
Adapter
approximators
Approximator
CompositionalApproximator
ContinuousApproximator
EnsembleApproximator
ModelComparisonApproximator
RatioApproximator
ScoringRuleApproximator
augmentations
nnpe
NNPE
datasets
DiskDataset
EnsembleDataset
EnsembleIndexedDataset
EnsembleOnlineDataset
OfflineDataset
OnlineDataset
diagnostics
metrics
accuracy
accuracy_random_points
bootstrap_comparison
brier_score
calibration_error
calibration_log_gamma
classifier_two_sample_test
correlation
expected_calibration_error
gamma_discrepancy
gamma_null_distribution
posterior_contraction
posterior_z_score
root_mean_squared_error
summary_space_comparison
plots
bayes_factor_recovery
calibration_ecdf
calibration_ecdf_from_quantiles
calibration_histogram
coverage
loss
mc_calibration
mc_confusion_matrix
mmd_hypothesis_test
pairs_posterior
pairs_quantity
pairs_samples
pairwise_bayes_factors
plot_quantity
recovery
recovery_from_estimates
z_score_contraction
distributions
DiagonalNormal
DiagonalStudentT
Distribution
Mixture
experimental
AutoEncoder
FreeFormFlow
LatentInferenceNetwork
VariationalAutoEncoder
links
CholeskyFactor
Leaky
Ordered
OrderedQuantiles
PositiveDefinite
metrics
functional
maximum_mean_discrepancy
root_mean_squared_error
MaximumMeanDiscrepancy
RootMeanSquaredError
networks
inference
consistency
ConsistencyModel
StableConsistencyModel
coupling
layers
permutations
transforms
ActNorm
CouplingFlow
InvertibleLayer
diffusion
find_noise_schedule
CosineNoiseSchedule
DiffusionModel
EDMNoiseSchedule
NoiseSchedule
flow_matching
FlowMatching
inference_network
InferenceNetwork
scoring
scoring_rules
ScoringRuleNetwork
PointNetwork
summary
ConvolutionalNetwork
DeepSet
FusionNetwork
FusionTransformer
SetTransformer
SummaryNetwork
TimeSeriesNetwork
TimeSeriesTransformer
subnets
DiffusionTransformer
MLP
ResidualUViT
TimeMLP
UNet
UViT
defaults
scoring_rules
BrierScore
CategoricalScoringRule
CrossEntropyScore
ExponentialScore
LogisticScore
MeanScore
MedianScore
MixtureScore
MvNormalScore
NormedDifferenceScore
ParametricDistributionScore
PolynomialScore
QuantileScore
ScoringRule
simulators
make_simulator
BernoulliGLM
BernoulliGLMRaw
GaussianLinear
GaussianLinearUniform
GaussianMixture
HierarchicalSimulator
InverseKinematics
LambdaSimulator
LotkaVolterra
ModelComparisonSimulator
SIR
SLCP
SLCPDistractors
SequentialSimulator
Simulator
TwoMoons
types
utils
keras_utils
call_accepts_kwarg
inverse_shifted_softplus
inverse_softplus
logits_relative_to_target
resolve_seed
shifted_softplus
logging
critical
debug
error
exception
format_duration
info
log
warn_once
warning
numpy_utils
credible_interval
inverse_shifted_softplus
inverse_sigmoid
inverse_softplus
one_hot
shifted_softplus
softplus
rng
next_seed_sequence
next_uint32
reseed_generator
reseed_random_state
serialization
deserialize
deserialize_value_or_type
serializable
serialize
serialize_value_or_type
tree
flatten_shape
add_metric
add_titles_and_labels
batched_call
calibration_curve
call_accepts_kwarg
check_estimates_prior_shapes
check_lengths_same
compute_empirical_coverage
concatenate_valid
concatenate_valid_shapes
confusion_matrix
convert_args
convert_kwargs
deserialize_value_or_type
detailed_loss_callback
devices
dim_maybe_nested
expand
expand_as
expand_left
expand_left_as
expand_left_to
expand_right
expand_right_as
expand_right_to
expand_tile
expand_to
feature_mask
fill_triangular_matrix
filter_kwargs
find_batch_size
find_distribution
find_inference_network
find_memory_budget
find_network
find_permutation
find_pooling
find_recurrent_net
find_scoring_rule
find_summary_network
find_transform
format_bytes
format_duration
integrate
integrate_stochastic
jacobian
jacobian_trace
jvp
layer_kwargs
linsolve_batched
make_quadratic
maybe_mask_tensor
model_kwargs
next_seed_sequence
next_uint32
non_batch_axis
optimal_transport
pad
parse_bytes
pickle_load
positive_diag
prepare_plot_data
prettify_subplots
repeat_and_flatten
reseed_generator
reseed_random_state
resolve_seed
sample_input_masks
searchsorted
sequential_kwargs
serialize_value_or_type
simultaneous_ecdf_bands
size_of
slice_maybe_nested
split_arrays
split_tensors
squeeze_inner_estimates_dict
stack_valid
tile_axis
tree_concatenate
tree_stack
vjp
weighted_mean
MaskName
workflows
BasicWorkflow
CompositionalWorkflow
EnsembleWorkflow
ModelComparisonWorkflow
bayesflow.wrappers
mamba
mamba
Mamba
mamba_block
MambaBlock
API Reference
utils
non_batch_axis
non_batch_axis
#
bayesflow.utils.
non_batch_axis
(
x
:
Tensor
)
[source]
#
On this page
non_batch_axis()
Edit on GitHub
