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)

pytorch特定操作，用于查找张量列表的尺寸明智均值

1 个答案: