Type - | -Definition - | -Notes - | -
Tensor - | -Pytorch Tensor type - | -https://pytorch.org/docs/stable/tensors.html#torch.Tensor - | -
Scalar - | -Any numerical types from Python, including integral types, floating point types, and zero dimensional tensors. - | -float and int argument types should suffice for most algorithms (you should only use Scalar if the operator truly may accept either type) - | -
int - | -python int (binded as int64_t in C++) - | -- | -
float - | -python float (binded as double in C++) - | -- | -
bool - | -python bool - | -- | -
str - | -python string - | -- | -
ScalarType - | -Enum type for all permissible dtype - | -https://pytorch.org/docs/stable/tensor_attributes.html#torch.dtype - | -
Layout - | -Enum type for all permissible Layout - | -https://pytorch.org/docs/stable/tensor_attributes.html#torch.layout - | -
MemoryFormat - | -Enum type for all permissible MemoryFormat - | -https://pytorch.org/docs/stable/tensor_attributes.html#torch-memory-format - | -
Device - | -torch.device -> str - | -https://pytorch.org/docs/stable/tensor_attributes.html#torch.device - | -
Tuple - | -Python tuple - | -- | -
List - | -Python list - | -- | -
Dict - | -Python dict with keys Scalar as defined in the table above. - | -- | -
Format - | -Column Order - | -
contiguous_format - | -NCHW - | -
Channels_last - | -NHWC - | -
(no name) - | -Other permutations - | -
-Op Set - | -Canonical ATen - | -Custom Op - | -|
ATen - | -Allowed - | -Allowed, must have meta kernel - | -- | -
Edge - | -Aten + Type specializations - | -Allowed - | -- | -
{description}
-{siteConfig.tagline}
-