Class Medium3DShapes

tests/datasets/ShapeDatasets.h:380–387

Data set containing medium 3D tensor shapes. */

/** Data set containing medium 3D tensor shapes. */
class Medium3DShapes final : public ShapeDataset
{
public:
    Medium3DShapes()
        : ShapeDataset("Shape", {TensorShape{42U, 37U, 8U}, TensorShape{57U, 60U, 13U}, TensorShape{83U, 72U, 14U}})
    {
    }
};

/** Data set containing medium 4D tensor shapes. */
class Medium4DShapes final : public ShapeDataset