如何查看一个tensor对象的shape
时间: 2024-03-10 20:48:34 浏览: 102
基于tf.shape(tensor)和tensor.shape()的区别说明
在 PyTorch 中,你可以使用 `shape` 属性或 `size()` 函数来查看一个 Tensor 对象的形状。例如,如果你有一个名为 `tensor` 的 Tensor 对象,你可以使用以下代码查看它的形状:
```python
print(tensor.shape)
# 或者
print(tensor.size())
```
上述代码将输出一个元组,其中包含 Tensor 对象的形状信息。例如,如果 `tensor` 的形状是 `(3, 4, 5)`,那么上述代码将输出 `(3, 4, 5)`。
注意,`shape` 和 `size()` 函数的返回值是元组,而不是列表,因此你不能修改 Tensor 对象的形状。如果你想要修改 Tensor 对象的形状,你需要使用 `view()` 函数或者 `reshape()` 函数。
阅读全文