pytorch_nndct.InputSpec - 1.3 Japanese

モジュールへの各入力の dtype と形状を指定します。

引数

InputSpec(shape, dtype)

shape

形状タプル、想定される入力の形状。

dtype

想定される入力の torch.dtype。