pyvene.models.modeling_utils.bhsd_to_bs_hd

Contents

pyvene.models.modeling_utils.bhsd_to_bs_hd#

bhsd_to_bs_hd(tensor)[source]#

Convert a tensor of shape (b, h, s, d) to (b, s, h*d).