pyvene.models.modeling_utils.b_sd_to_bsd

Contents

pyvene.models.modeling_utils.b_sd_to_bsd#

b_sd_to_bsd(tensor, s)[source]#

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