tfmri.sampling.flatten_trajectory

flatten_trajectory(trajectory)[source]

Flatten a trajectory tensor.

Parameters

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

Returns

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