class NDimensionalSoftmax(Softmax): decorators = [WithExtraDims()]
class NDimensionalSoftmaxWithMask(SoftmaxWithMask): decorators = [WithExtraDims()]
class SoftmaxWithExtraDims(Softmax): decorators = [WithExtraDims()]
class LinearWithExtraDims(Linear): decorators = [WithExtraDims()]
class SelectInEachSubtensor(SelectInEachRow): decorators = [WithExtraDims()]
class NDimensionalCostObject(CostObject): decorators = [WithExtraDims()]