我有一个有向量输出的神经网络Network。我不使用典型的损失函数,而是实现自己的损失函数,这是类中的一种方法。这看起来像是: def __init__(self, network, optimizer): self.optimizer = optimizer def cost_function(relevant_data):
...implementation of cost function with respect to outpu
例如,我有一个N B x H张量列表(即N x B x H张量)和一个N向量列表(即N x B张量)。我要把列表中的每个B x H张量乘以相应的B维张量,从而得到N x H张量。我知道如何使用单个for-loop和PyTorch来实现计算,但是有矢量注入吗?(即不使用for-loop,只使用PyTorch/numpy操作)