modelzoo.common.pytorch.utils.to_tensor#

modelzoo.common.pytorch.utils.to_tensor(value, device=None)[source]#

If the provided value is a Python int or float, it converts them into PyTorch Tensors of type int32 and float32 respectively. Otherwise, it just returns the value.

previous

modelzoo.common.pytorch.utils.to_cpu

next

modelzoo.common.pytorch.utils.visit_structure