pytorch_nndct.InputSpec - 1.3 Japanese

Vitis AI オプティマイザー ユーザー ガイド (UG1333)

Document ID
UG1333
Release Date
2021-02-03
Version
1.3 Japanese

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

引数

InputSpec(shape, dtype)
shape
形状タプル、想定される入力の形状。
dtype
想定される入力の torch.dtype。