我有一个形状张量,(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,但我不知道如何使用开始和结束索引,而且文档并不清晰。
事实证明,tensorflow 不需要您使用 tf.slice,因为您可以简单地使用 numpy 切片:
first_half = input[:16]
second_half = input[16:]
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
1519 次 |
| 最近记录: |