set_data_type
Applicability
Product |
Supported |
|---|---|
Atlas 350 Accelerator Card |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
Function Usage
Sets the tensor data type.
Prototype
1 | set_data_type(data_type: DataType) -> Tensor |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
data_type |
Input |
Data type to be set. For details, see DataType. |
Returns
(Tensor) Returns itself to support chain calls.
Restrictions
- If data_type is not of the DataType, TypeError is thrown.
- If the setting fails, RuntimeError is thrown.
Parent topic: Tensor