tf.range详解

官网链接
https://tensorflow.google.cn/api_docs/python/tf/range

tf.range(limit, delta=1, dtype=None, name='range')
tf.range(start, limit, delta=1, dtype=None, name='range')

以start为起点,delta为变化量,生成不超过,且不包含limit的Tensor
例子

import tensorflow as tf

with tf.Session() as sess:
    # tf.range(start, limit, delta)
    a = tf.range(3, 18, 3)
    print(sess.run(a))
    b = tf.range(3, 1, -0.5)
    print(sess.run(b))
    c = tf.range(5)
    print(sess.run(c))
[ 3  6  9 12 15]
[3.  2.5 2.  1.5]
[0 1 2 3 4]

猜你喜欢

转载自blog.csdn.net/lllxxq141592654/article/details/85395386