fix

Author: yueyinqiu

RELEASENOTES.md

@@ -8,8 +8,15 @@ __Breaking Changes__:
 
 __API Changes__:
 
+- A new option `autoDispose` has been added into `DataLoader`s, which indicates whether to dispose the collated tensors before the next iteration.
+    - The default collate functions will now always dispose the intermediate tensors, rather than wait for the next iteration.
+- A new abstract method `DisposeTensor` has been added to `Dataset<>`s.
+    - This method will be used by `DataLoader`, to dispose the values provided by `GetTensor`.
+    - `Dataset` and `IterableDataset` has implemented this method, so if your dataset is inherited from them, please check whether the disposal should be avoided in your case.
+
 __Bug Fixes__:
 
+- `TensorDataset` will now keep the aliases, detached from dispose scopes, to avoid the unexpected disposal.
 
 # NuGet Version 0.102.4
 

src/TorchAudio/Datasets/SpeechCommandsDataset.cs

@@ -48,6 +48,11 @@ internal SpeechCommandsDataset(string path, string subset)
 
                 public override long Count => _walker.LongLength;
 
+                public override void DisposeTensor(SpeechCommandsDatasetItem tensor)
+                {
+                    tensor.waveform.Dispose();
+                }
+
                 public override SpeechCommandsDatasetItem GetTensor(long index)
                 {
                     var audioPath = _walker[index];

src/TorchAudio/Datasets/YesnoDataset.cs

@@ -22,6 +22,11 @@ public YesnoDataset(string directoryPathInArchive)
 
                 public override long Count => audioPathList.LongLength;
 
+                public override void DisposeTensor(YesnoDatasetItem tensor)
+                {
+                    tensor.waveform.Dispose();
+                }
+
                 public override YesnoDatasetItem GetTensor(long index)
                 {
                     var (waveform, sample_rate) = torchaudio.load(audioPathList[index]);

src/TorchSharp/Utils/TensorDataset.cs

@@ -37,7 +37,7 @@ internal TensorDataset(torch.Tensor[] tensors)
                 long size1 = tensors[0].shape[0];
                 if (!tensors.All(t => t.shape[0] == size1)) throw new ArgumentException("All tensors must have the same first dimension size.", nameof(tensors));
 
-                _tensors = tensors.Select(x => x.alias()).ToArray();
+                _tensors = tensors.Select(x => x.alias().DetachFromDisposeScope()).ToArray();
             }
 
             /// <summary>