Module: TorchRec::Modules::Utils

Defined in:: lib/torchrec/modules/utils.rb

Class Method Summary collapse

.extract_module_or_tensor_callable(module_or_callable) ⇒ Object

TODO.

Class Method Details

.extract_module_or_tensor_callable(module_or_callable) ⇒ `Object`

TODO



6
7
8

# File 'lib/torchrec/modules/utils.rb', line 6

def extract_module_or_tensor_callable(module_or_callable)
  module_or_callable
end