Class SoftmaxLayer4DShapes

tests/datasets/ShapeDatasets.h:931–942

Data set containing large and small softmax layer 4D shapes. */

/** Data set containing large and small softmax layer 4D shapes. */
class SoftmaxLayer4DShapes final : public ShapeDataset
{
public:
    SoftmaxLayer4DShapes()
        : ShapeDataset("Shape",
                       {
                           TensorShape{9U, 9U, 9U, 9U},
                           TensorShape{31U, 10U, 1U, 9U},
                       })
    {
    }
};

/** Data set containing 2D tensor shapes relative to an image size. */
class SmallImageShapes final : public ShapeDataset