Datatset class multiple input in pytorch(closing as solved)

[asha97]

To include multiple data at once, for image I thing concatenation after flattening can be done,