🤗 PyTorch如何查看数据类型和大小？ 📏
发布时间：2025-03-26 19:14:21来源：
在使用PyTorch进行深度学习项目时，了解数据的类型和尺寸非常重要。首先，查看数据类型可以用`.dtype`属性，例如：
```python
tensor = torch.tensor([1, 2, 3])
print(tensor.dtype) 输出：torch.int64
```
接着，若想查看数据的形状（即大小），可以使用`.shape`属性。比如：
```python
print(tensor.shape) 输出：torch.Size([3])
```
如果涉及的是PyTorch的数据集（Dataset），可以通过以下方式获取数据集大小：
```python
dataset = torchvision.datasets.MNIST(root='./data', download=True)
print(len(dataset)) 输出数据集长度，如60000
```
此外，如果你需要查看单个样本的大小，可以结合索引读取：
```python
sample, label = dataset[0]
print(sample.shape) 输出样本形状，如torch.Size([1, 28, 28])
```
掌握这些方法后，无论是调试还是优化模型，都能更加得心应手！💡
免责声明：本答案或内容为用户上传，不代表本网观点。其原创性以及文中陈述文字和内容未经本站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺，请读者仅作参考，并请自行核实相关内容。 如遇侵权请及时联系本站删除。
