cSAXS/cdtools
0
0 Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-11-09 15:54:02 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove last of t.tensor usage from dataset loading

Browse Source
This commit is contained in:
Abe Levitan committed 2019-10-04 17:00:43 -04:00
1 parent 120e23ef15
commit e987d40df8
2 files changed
+4 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

CDTools/datasets/ptycho_2d_dataset.py
+2 -2
View File
@@ -62,8 +62,8 @@ class Ptycho2DDataset(CDataset):
super(Ptycho2DDataset,self).__init__(*args, **kwargs)
self.axes = copy(axes)
self.translations = t.tensor(translations)
self.patterns = t.tensor(patterns)
self.translations = t.Tensor(translations).clone()
self.patterns = t.Tensor(patterns).clone()
if self.mask is None:
self.mask = t.ones(self.patterns.shape[-2:]).to(dtype=t.bool)
self.mask.masked_fill_(t.isnan(t.sum(self.patterns,dim=(0,))),0)