set_data_type

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.

Constraints

  • If data_type is not of the DataType type, TypeError is thrown.
  • If the setting fails, RuntimeError is thrown.