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.

参数: element_type – [in] The element type of the TensorType.

TypePtr element() const

Get the element type of this TensorType.

返回: The element type of this TensorType.