tfmri.sampling.flatten_density

flatten_density(density)[source]

Flatten a density tensor.

Parameters

density – A Tensor. Must have shape [..., views, samples].

Returns

A reshaped Tensor with shape [..., views * samples].