torch输出张量的大小

可以通过张量的属性`size()`或者`shape`来输出张量的大小，例如：

import torch

# 创建一个大小为(2, 3, 4)的三维张量
x = torch.randn(2, 3, 4)

# 输出张量的大小
print(x.size())
print(x.shape)

输出结果为：

torch.Size([2, 3, 4])
torch.Size([2, 3, 4])

相关问题

torch输出张量的维度

可以通过张量的属性`dim()`或者`ndim`来输出张量的维度，例如：

import torch

# 创建一个大小为(2, 3, 4)的三维张量
x = torch.randn(2, 3, 4)

# 输出张量的维度
print(x.dim())
print(x.ndim)

输出结果为：

3
3

torch打印张量的类型

可以通过张量的属性`dtype`来打印张量的类型，例如：

import torch

# 创建一个大小为(2, 3)的整型张量
x = torch.randint(10, (2, 3))

# 打印张量的类型
print(x.dtype)

输出结果为：

torch.int64

也可以通过`type()`函数来打印张量的类型，例如：

import torch

# 创建一个大小为(2, 3)的整型张量
x = torch.randint(10, (2, 3))

# 打印张量的类型
print(type(x))

输出结果为：

<class 'torch.Tensor'>