def softmax(self):
numers = torch.exp(Tensor([self.weight @ self.feature(self.state, a) for a in range(self.action_size)]))
return numers / sum(numers)
微信
公众号
扫码关注公众号