求张量表的维平均值的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)