Class Small3DShapes

tests/datasets/ShapeDatasets.h:103–118

Data set containing small 3D tensor shapes. */

/** Data set containing small 3D tensor shapes. */
class Small3DShapes final : public ShapeDataset
{
public:
    Small3DShapes()
        : ShapeDataset("Shape",
                       {
                           TensorShape{1U, 7U, 7U},
                           TensorShape{2U, 5U, 4U},

                           TensorShape{7U, 7U, 5U},
                           TensorShape{16U, 16U, 5U},
                           TensorShape{27U, 13U, 37U},
                       })
    {
    }
};

/** Data set containing tiny 4D tensor shapes. */
class Tiny4DShapes final : public ShapeDataset