unisim.backend.motrix.tensor.state_views

unisim.backend.motrix.tensor.state_views(backend, fields, device)[source]

Copy requested state blocks to one Torch device in one packed H2D.

Parameters:
Return type:

dict[str, Any]