一个4 * 600 * 600的tensor, 如何切成一个 3 * 600 * 600的tensor,python
时间: 2023-09-12 16:06:02 浏览: 81
python将一张图切为3*6张小图
可以使用numpy中的切片操作实现:
```python
import numpy as np
tensor_4d = np.random.rand(4, 600, 600)
tensor_3d = tensor_4d[:3, :, :]
```
其中,`tensor_4d[:3, :, :]`表示从第0维开始(即第一个维度),取到第3个维度(不包含第3个维度),其他维度全部取(即第1维和第2维)。这样就得到了一个3 * 600 * 600的tensor。
阅读全文