Generate a SoftMax operator
| 1462 | |
| 1463 | |
| 1464 | class SoftMax(Layer): |
| 1465 | """ |
| 1466 | Generate a SoftMax operator |
| 1467 | """ |
| 1468 | |
| 1469 | def __init__(self): |
| 1470 | super(SoftMax, self).__init__() |
| 1471 | |
| 1472 | def forward(self, x): |
| 1473 | return autograd.softmax(x) |
| 1474 | |
| 1475 | |
| 1476 | class MeanSquareError(Layer): |