在PyTorch中，labels.size(0)返回的是一个张量的第一维大小，即批次中样本的数量。这在训练分类器时用于计算损失函数和更新模型参数。
labels是一个尺寸为 的张量[N, 1]，其中N等于批次中的样本数。返回具有张量维度的.size(...)元组 ( ) 的子类，并返回具有第一个（基于 0 的）维度值的整数（即）。torch.Size.size(0)N
https://stackoverflow.com/questions/55565687/i-dont-understand-the-code-for-training-a-classifier-in-pytorch
最低0.47元/天 解锁文章
扫一扫
评论 2
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
labels是一个尺寸为 的张量[N, 1]，其中N等于批次中的样本数。返回具有张量维度的.size(...)元组 ( ) 的子类，并返回具有第一个（基于 0 的）维度值的整数（即）。torch.Size.size(0)N
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
