在 tensorflow 中将张量切成两半

Qub*_*bix 0 slice tensorflow

我有一个形状张量,(32, 32, 32, 1)我想沿着第一维将它切成两个张量,包含上半部分和下半部分

half1  with shape = (16, 32, 32, 1)
half2  with shape = (16, 32, 32, 1)
Run Code Online (Sandbox Code Playgroud)

我正在尝试使用 tf.slice,但我不知道如何使用开始和结束索引,而且文档并不清晰。

Qub*_*bix 5

事实证明,tensorflow 不需要您使用 tf.slice,因为您可以简单地使用 numpy 切片:

first_half = input[:16]
second_half = input[16:]
Run Code Online (Sandbox Code Playgroud)