Method cast

src/compute-engine/tensor/tensor-fields.ts:71–112

(
    x: number | number[],
    dtype: TensorDataType
  )

  cast(x: number, dtype: 'expression'): undefined | Expression;
  cast(x: number[], dtype: 'float64'): undefined | number[];
  cast(x: number[], dtype: 'float32'): undefined | number[];
  cast(x: number[], dtype: 'int32'): undefined | number[];
  cast(x: number[], dtype: 'uint8'): undefined | number[];
  cast(x: number[], dtype: 'complex128'): undefined | Complex[];
  cast(x: number[], dtype: 'complex64'): undefined | Complex[];
  cast(x: number[], dtype: 'bool'): undefined | boolean[];
  // cast(x: number[], dtype: 'string'): undefined | string[];
  cast(x: number[], dtype: 'expression'): undefined | Expression[];
  cast(
    x: number | number[],
    dtype: TensorDataType
  ):
    | undefined
    | Complex
    | number
    | boolean
    // | string
    | Expression
    | Complex[]
    | number[]
    | boolean[]
    // | string[]
    | Expression[] {
    const ce = this.ce;
    switch (dtype) {
      case 'float64':
      case 'float32':
      case 'int32':
      case 'uint8':
        return x;
      case 'complex128':
      case 'complex64':
        return Array.isArray(x)
          ? x.map((x) => ce.complex(x))
          : this.ce.complex(x);
      case 'bool':
        return Array.isArray(x)
          ? x.map((x) => (x === 0 ? false : true))
          : x === 0
            ? false
            : true;
      // case 'string':
      //   return Array.isArray(x)
      //     ? x.map((x) => Number(x).toString())
      //     : Number(x).toString();