Searched defs:_apply_to_tensors (Results 1 – 1 of 1) sorted by relevance
218 def _apply_to_tensors(fn: Callable[[torch.Tensor], Q], container: torch.Tensor) -> Q: function223 def _apply_to_tensors(fn: Callable[[torch.Tensor], Any], container: R) -> R: function227 def _apply_to_tensors(fn, container): function