how to get pytorch tensor shape
时间: 2024-03-20 07:38:43 浏览: 114
lamp-cloud 基于jdk21、jdk17、jdk8 + SpringCloud + SpringBoot 开发的微服务中后台快速开发平台,专注于多租户(SaaS架构)解决方案
可以通过PyTorch Tensor的`shape`属性来获取其形状信息,例如:
```python
import torch
# 创建一个3行2列的Tensor
x = torch.Tensor([[1, 2], [3, 4], [5, 6]])
# 获取x的形状
print(x.shape)
```
输出:
```
torch.Size([3, 2])
```
这里的`torch.Size([3, 2])`表示`x`是一个3行2列的Tensor。
阅读全文