Class Medium2DShapes

tests/datasets/ShapeDatasets.h:368–377

Data set containing medium 2D tensor shapes. */

/** Data set containing medium 2D tensor shapes. */
class Medium2DShapes final : public ShapeDataset
{
public:
    Medium2DShapes()
        : ShapeDataset("Shape",
                       {TensorShape{42U, 37U}, TensorShape{57U, 60U}, TensorShape{128U, 64U}, TensorShape{83U, 72U},
                        TensorShape{40U, 40U}})
    {
    }
};

/** Data set containing medium 3D tensor shapes. */
class Medium3DShapes final : public ShapeDataset