pytorch ----【输入张量.data.size()/输入张量.size()】的使用

import torch
test_data = torch.randn(1, 12, 32, 32)
batchsize, num_channels, height, width = test_data.data.size()

在提供的代码中，test_data 是一个形状为 (1, 12, 32, 32) 的随机张量，表示一个批次（batch）中有 1 张图像，每张图像有 12 个通道，图像的高度和宽度均为 32 像素。

注意事项

• test_data.size() 返回一个包含张量各个维度大小的元组，可以直接解包到多个变量中。
• 确保在调用 .size() 时，使用的是 size() 而不是 data.size()，后者在新的 PyTorch 版本中已不推荐使用。

打印内容：

print(f"Batch Size: {batchsize}, Channels: {num_channels}, Height: {height}, Width: {width}")
#Batch Size: 1, Channels: 12, Height: 32, Width: 32