tf-keras-skeleton.py 文件源码

def _to_tensor(x, dtype):
      """Convert the input `x` to a tensor of type `dtype`.

      Arguments:
          x: An object to be converted (numpy array, list, tensors).
          dtype: The destination type.

      Returns:
          A tensor.
      """
      x = ops.convert_to_tensor(x)
      if x.dtype != dtype:
        x = math_ops.cast(x, dtype)
      return x