Python ``tensorflow.IndexedSlices.shape`` 属性与实例


发布日期 : 2024-03-25 11:47:56 UTC

访问量: 9 次浏览

Python – tensorflow.IndexedSlices.shape 属性

TensorFlow是谷歌设计的开源Python库,用于开发机器学习模型和深度学习神经网络。

shape是用来获取代表密集张量形状的 tensorflow.TensorShape 的。

语法:
tensorflow.IndexedSlices.shape

返回:它返回 tensorflow.TensorShape ,代表密集张量的形状。

示例 1:

# Importing the library
import tensorflow as tf

# Initializing the input
data = tf.constant([[1, 2, 3], [4, 5, 6]])

# Printing the input
print('data: ', data)

# Calculating result
res = tf.IndexedSlices(data, [0], tf.constant([1, 2]))

# Finding Shape
shape = res.shape

# Printing the result
print('Shape: ', shape)

输出:

data: tf.Tensor(
[[1 2 3]
[4 5 6]], shape=(2, 3), dtype=int32)
Shape: (1, 2)

示例 2:

# Importing the library
import tensorflow as tf

# Initializing the input
data = tf.constant([[1, 2, 3], [4, 5, 6]])

# Printing the input
print('data: ', data)

# Calculating result
res = tf.IndexedSlices(data, [0], tf.constant([1]))

# Finding Shape
shape = res.shape

# Printing the result
print('Shape: ', shape)

输出:

data: tf.Tensor(
[[1 2 3]
[4 5 6]], shape=(2, 3), dtype=int32)
Shape: (1, )