在不複製資料的情況下將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...