tensorflow 展开a，B = t1[1：3]

bqucvtff 于 2023-11-21 发布在其他

关注(0)|答案(1)|浏览(123)

这是否保证在未来的版本中也能正常工作？

import tensorflow as tf

t1 = tf.constant([0, 1, 2, 3, 4, 5, 6, 7])
a, b = t1[1:3]
print(f"a:{a} b:{b}")
---
a:1 b:2

字符串
不确定这是Python功能，将来可能无法使用，还是Tensorflow作为内置功能支持的功能。

1条答案

是的，这是文件的一部分。
使用tf.slice执行类似NumPy的Tensor切片。

t1 = tf.constant([0, 1, 2, 3, 4, 5, 6, 7])

print(tf.slice(t1,
               begin=[1],
               size=[3]))

字符串
或者，您可以使用更Python化的语法。请注意，Tensor切片在开始-停止范围内均匀分布。

print(t1[1:4])

型