tensorflow2 1的資料型別轉換

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

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...