get_plan_attribute_dtype#

nvmath.bindings.cutensor.get_plan_attribute_dtype(int attr)[source]#

Get the Python data type of the corresponding PlanAttribute attribute.

Parameters:

attr (PlanAttribute) – The attribute to query.

Returns:

The data type of the queried attribute.

Note

This API has no C counterpart and is a convenient helper for allocating memory for plan_get_attribute().