ExTorch.NN.Introspect.Schema (extorch v0.4.0)

Structured representation of a JIT model's architecture.

Summary

t()

@type param_info() :: %{
  name: String.t(),
  shape: [integer()],
  dtype: atom(),
  requires_grad: boolean()
}

@type submodule_info() :: %{
  name: String.t(),
  type_name: String.t(),
  parameters: [param_info()]
}

@type t() :: %ExTorch.NN.Introspect.Schema{
  methods: [String.t()],
  parameters: [param_info()],
  submodules: [submodule_info()]
}