将张量转换为 int32 类型

最新推荐文章于 2025-06-18 16:58:08 发布

原创最新推荐文章于 2025-06-18 16:58:08 发布 · 6.2k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#tf.to_int32 #python

python 专栏收录该内容

58 篇文章

订阅专栏

本文详细介绍了TensorFlow中tf.to_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))