autocast#

torchsim.base.autocast(func)[source]#

Force all inputs to be torch tensors of the same size on the same device.

Parameters:

func (Callable)

Return type:

Callable

Examples using torchsim.base.autocast#

Custom Simulator

Custom Simulator