tensorflow2 1的資料型別轉換

2021-10-04 09:20:15 字數 1455 閱讀 7503

在不複製資料的情況下將tensor從一種資料型別轉換為另一種型別。

tf.bitcast

(input

,type

, name=

none

)

引數說明:

input- 配置需要進行型別轉換的tensor,tensor的型別可以為:bfloat16, half, float32, float64, int64, int32, uint8, uint16, uint32, uint64, int8, int16, complex64, complex128, qint8, quint8, qint16, quint16, qint32.

type- 配置轉換後的資料型別,可以選擇的型別包括:tf.bfloat16, tf.half, tf.float32, tf.float64, tf.int64, tf.int32, tf.uint8, tf.uint16, tf.uint32, tf.uint64, tf.int8, tf.int16, tf.complex64, tf.complex128, tf.qint8, tf.quint8, tf.qint16, tf.quint16, tf.qint32.

name:配置運算操作的名稱。

示例**:

import tensorflow as tf

a = tf.constant([32

.,25.

])b = tf.bitcast(a,

type

= tf.int32)

print

(a.dtype, b.dtype)

'float32'

>

'int32'

>

tf.cast()函式的作用是執行 tensorflow 中張量資料型別轉換,比如讀入的如果是int8型別的,一般在要在訓練前把影象的資料格式轉換為float32。

用法:

tf.cast(

x,dtype,

name=

none

)

引數說明:

示例:float轉換為bool:

import tensorflow as tf

a = tf.variable([1

,0,0

,1,1

])b = tf.cast(a,dtype=tf.

bool)b

out[87]

:,), dtype=

bool

, numpy=array(

[true

,false

,false

,true

,true])

>

tensorflow2 1安裝指南

開啟anconda prompt 建立conda虛擬環境 用create n 新建乙個名叫tf2.1的環境用python3.7版本 conda create n tf2.1 python 3.7 進入tensorflow2.1環境 conda activate tf2.1 安裝英偉達的sdk10.1...

tensorflow2 1的維度變換

函式的作用是將tensor變換為引數shape的形式,其中shape為乙個列表形式,特殊的一點是列表中可以存在 1 1代表的含義是不用我們自己指定這一維的大小,函式會自動計算,但列表只能存在乙個 1。如果存在多個 1,就是乙個存在多解的方程 a tf.random.normal 4 28,28 3 ...

TensorFlow2 1張量排序

排序函式tf.sort 用法 tf.sort values,axis 1 direction ascending name none 引數說明 排序的座標tf.argsort 返回張量的索引,該張量給出沿軸的排序順序。用法 tf.argsort values,axis 1 direction asc...