1. 程式人生 > >將張量轉換為 int32 型別

將張量轉換為 int32 型別

tf.to_int32函式:

tf.to_int32(
    x,
    name='ToInt32'
)

函式引數:

  • x:一個 Tensor 、SparseTensor、list或ndarray
  • name:操作的名稱(可選)。

函式返回值:

tf.to_int32函式返回一個 Tensor 或 SparseTensor,與 x (型別為 int32)具有相同的形狀。

例子:

import numpy as np
import tensorflow as tf

a = [[1,2,3], [2,3,4]]
b = np.array([[1,2,3], [2,3,4]])
c = tf.convert_to_tensor([[1,2,3], [2,3,4]])
d = tf.to_int32(a)
e = tf.to_int32(b)
f = tf.to_int32(c)
sess = tf.InteractiveSession()
print(sess.run(d))
print(sess.run(e))
print(sess.run(f))