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
原文地址：https://blog.csdn.net/m0_48241022/article/details/142483995
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！