python - Pytorch 特定操作，用于查找张量列表的维度均值

Question

我有一个张量列表，每个维度为 300x1，我需要根据列表找到每个维度的均值和方差。 这到底是如何实现的？。我可以使用 numpy 数组在 numpy 中完成这项工作，使用以下代码片段:

for item in embeddingLists:    # embeddingLists is a list of lists
                               # item contains list of numpy arrays
    tempVal =  np.mean(np.array(item),axis=0)
    meanVects.append(tempVal)

    temVar = np.var(np.array(item),axis=0,ddof=1) 
    varVects.append(temVar)

Answer 1

在 pytorch 论坛的帮助下，我可以解决这个问题。 Link to discussion

相关代码为:

for item in embeddingLists:       
    tempItem = [stuff.unsqueeze(0)  for stuff in item] #convert a 1x300 tensor
    coomn = torch.cat(tempItem)     # Convert to a 12x300 tensor        
    temMean = torch.mean(coomn,dim=0)
    meanVects.append(temMean)
    temVar = torch.var(coomn,dim=0)
    varVects.append(temVar)