AttributeError: 'torch.Size' object has no attribute 'rank'

这错误提示是因为在 PyTorch 中，torch.Size 对象没有 rank 属性。相反，它有一个叫做 dim 的属性，它返回张量的维度数。因此，如果你想获取张量的维度数，你应该使用张量的 size() 方法，而不是访问 torch.Size 对象的 rank 属性。