test data loader for ConcatDataset

Author: yueyinqiu

test/TorchSharpTest/TestDataLoader.cs

@@ -4,6 +4,7 @@
 using System.Linq;
 using TorchSharp.Modules;
 using Xunit;
+using static TorchSharp.torch;
 
 
 namespace TorchSharp
@@ -306,5 +307,17 @@ public void ConcatDatasetTest()
             Assert.True(dataset2.Disposed);
             Assert.True(dataset3.Disposed);
         }
+
+        [Fact]
+        public void ConcatDatasetDataLoader()
+        {
+            using var dataset = torch.utils.data.ConcatDataset<IReadOnlyDictionary<string, Tensor>>(
+                new[] {
+                    new TestDataset(),
+                    new TestDataset(),
+                });
+            var dataloader = torch.utils.data.DataLoader(dataset, 10, false);
+            Assert.Equal(2, dataloader.Count);
+        }
     }
 }