tf.ones_like()

最新推荐文章于 2022-10-14 09:21:39 发布

原创最新推荐文章于 2022-10-14 09:21:39 发布 · 207 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

tf 专栏收录该内容

40 篇文章

订阅专栏

本文介绍了如何使用TensorFlow中的ones_like和zeros_like函数创建与输入张量形状和类型相同的全零和全一矩阵，并通过实例展示了它们的用法和运行结果。

tf.ones_like(
    input, dtype=None, name=None
)

在这里插入图片描述

import tensorflow as tf
import numpy as np

# 生成一个tensor，内部数据随机产生
a = tf.convert_to_tensor(np.random.random([2, 4, 5]), dtype=tf.float32)

# ones_like
b = tf.ones_like(a, dtype=tf.float32, name='ones_like')

# zeros_like
c = tf.zeros_like(a, dtype=tf.float32, name='zeros_like')

print(b)

print(c)

with tf.Session() as sess:
    b_, c_ = sess.run([b, c])
    print("b's shape: ", b_.shape)
    print("c's shape: ", c_.shape)
    print(b_)
    print(c_)