TorchModuleWrapper

Torch module wrapper layer for Keras.

Abstract Signature:

TorchModuleWrapper(module, name: str)

Keras

API: keras.layers.TorchModuleWrapper
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: keras.layers.TorchModuleWrapper
Strategy: Direct Mapping
Official Docs ↗