Class TensorType
Defined in File type.h
Inheritance Relationships
Base Type
public mindspore::api::Type
(Class Type)
Class Documentation
-
class TensorType : public mindspore::api::Type
TensorType defines the type of a tensor.
Public Functions
-
explicit TensorType(const TypePtr &element_type)
Construct TensorType from the given element type.
- Parameters
element_type – [in] The element type of the TensorType.
-
TypePtr element() const
Get the element type of this TensorType.
- Returns
The element type of this TensorType.
-
explicit TensorType(const TypePtr &element_type)