Class Small2DShapes

tests/datasets/ShapeDatasets.h:80–86

Data set containing small 2D tensor shapes. */

};
/** Data set containing small 2D tensor shapes. */
class Small2DShapes final : public ShapeDataset
{
public:
    Small2DShapes() : ShapeDataset("Shape", {TensorShape{1U, 7U}, TensorShape{5U, 13U}, TensorShape{32U, 64U}})
    {
    }
};

/** Data set containing tiny 3D tensor shapes. */
class Tiny3DShapes final : public ShapeDataset