docs
0.0-2dfa388-SNAPSHOT
API
Generated with
docs/torch/torch.data

torch.data

package torch.data

Members list

Type members

Classlikes

class DataLoader[Input, Batch](dataset: IndexedSeq[Input], batchSize: Int, shuffle: Boolean, collateFn: Seq[Input] => Batch) extends Iterable[Batch]

Provides an iterable over batches of a given dataset.

Provides an iterable over batches of a given dataset.

Attributes

Source
DataLoader.scala
Supertypes
trait Iterable[Batch]
trait IterableOps[Batch, Iterable, Iterable[Batch]]
trait IterableOnceOps[Batch, Iterable, Iterable[Batch]]
trait IterableOnce[Batch]
class Object
trait Matchable
class Any
Show all
Known subtypes
class ExampleDataLoader[D1, D2]
class TupleDataLoader[D1, D2]
final case class Example[D1 <: DType, D2 <: DType](feature: Tensor[D1], target: Tensor[D2])

Attributes

Source
Example.scala
Supertypes
trait Serializable
trait Product
trait Equals
class Object
trait Matchable
class Any
Show all
class ExampleDataLoader[D1 <: DType, D2 <: DType](dataset: IndexedSeq[Example[D1, D2]], batchSize: Int, shuffle: Boolean, collateFn: Seq[Example[D1, D2]] => (Tensor[D1], Tensor[D2])) extends DataLoader[Example[D1, D2], (Tensor[D1], Tensor[D2])]

Attributes

Source
DataLoader.scala
Supertypes
class DataLoader[Example[D1, D2], (Tensor[D1], Tensor[D2])]
trait Iterable[(Tensor[D1], Tensor[D2])]
trait IterableOps[(Tensor[D1], Tensor[D2]), Iterable, Iterable[(Tensor[D1], Tensor[D2])]]
trait IterableOnceOps[(Tensor[D1], Tensor[D2]), Iterable, Iterable[(Tensor[D1], Tensor[D2])]]
trait IterableOnce[(Tensor[D1], Tensor[D2])]
class Object
trait Matchable
class Any
Show all
trait TensorDataset[Input <: DType, Target <: DType] extends IndexedSeq[(Tensor[Input], Tensor[Target])]

Wraps a pair of tensors as a Seq.

Wraps a pair of tensors as a Seq.

Each sample will be retrieved by indexing tensors along the first dimension.

Attributes

Companion
object
Source
TensorDataset.scala
Supertypes
trait IndexedSeq[(Tensor[Input], Tensor[Target])]
trait IndexedSeqOps[(Tensor[Input], Tensor[Target]), IndexedSeq, IndexedSeq[(Tensor[Input], Tensor[Target])]]
trait IndexedSeq[(Tensor[Input], Tensor[Target])]
trait IndexedSeqOps[(Tensor[Input], Tensor[Target]), IndexedSeq, IndexedSeq[(Tensor[Input], Tensor[Target])]]
trait Seq[(Tensor[Input], Tensor[Target])]
trait SeqOps[(Tensor[Input], Tensor[Target]), IndexedSeq, IndexedSeq[(Tensor[Input], Tensor[Target])]]
trait Seq[(Tensor[Input], Tensor[Target])]
trait Equals
trait SeqOps[(Tensor[Input], Tensor[Target]), IndexedSeq, IndexedSeq[(Tensor[Input], Tensor[Target])]]
trait PartialFunction[Int, (Tensor[Input], Tensor[Target])]
trait Int => (Tensor[Input], Tensor[Target])
trait Iterable[(Tensor[Input], Tensor[Target])]
trait Iterable[(Tensor[Input], Tensor[Target])]
trait IterableFactoryDefaults[(Tensor[Input], Tensor[Target]), IndexedSeq]
trait IterableOps[(Tensor[Input], Tensor[Target]), IndexedSeq, IndexedSeq[(Tensor[Input], Tensor[Target])]]
trait IterableOnceOps[(Tensor[Input], Tensor[Target]), IndexedSeq, IndexedSeq[(Tensor[Input], Tensor[Target])]]
trait IterableOnce[(Tensor[Input], Tensor[Target])]
class Object
trait Matchable
class Any
Show all
Known subtypes
trait MNISTBase
class FashionMNIST
class MNIST
object TensorDataset

Attributes

Companion
trait
Source
TensorDataset.scala
Supertypes
class Object
trait Matchable
class Any
Self type
class TensorSeq[D <: DType](t: Tensor[D]) extends IndexedSeq[Tensor[D]]

Wraps a tensor as a Seq.

Wraps a tensor as a Seq.

Each sample will be retrieved by indexing tensors along the first dimension.

Value parameters

t

tensor to be wrapped as a seq

Attributes

Source
TensorSeq.scala
Supertypes
trait IndexedSeq[Tensor[D]]
trait IndexedSeq[Tensor[D]]
trait Seq[Tensor[D]]
trait Seq[Tensor[D]]
trait Equals
trait Int => Tensor[D]
trait Iterable[Tensor[D]]
trait Iterable[Tensor[D]]
trait IterableOnce[Tensor[D]]
class Object
trait Matchable
class Any
Show all
class TupleDataLoader[D1 <: DType, D2 <: DType](dataset: IndexedSeq[(Tensor[D1], Tensor[D2])], batchSize: Int, shuffle: Boolean, collateFn: Seq[(Tensor[D1], Tensor[D2])] => (Tensor[D1], Tensor[D2])) extends DataLoader[(Tensor[D1], Tensor[D2]), (Tensor[D1], Tensor[D2])]

Attributes

Source
DataLoader.scala
Supertypes
class DataLoader[(Tensor[D1], Tensor[D2]), (Tensor[D1], Tensor[D2])]
trait Iterable[(Tensor[D1], Tensor[D2])]
trait IterableOps[(Tensor[D1], Tensor[D2]), Iterable, Iterable[(Tensor[D1], Tensor[D2])]]
trait IterableOnceOps[(Tensor[D1], Tensor[D2]), Iterable, Iterable[(Tensor[D1], Tensor[D2])]]
trait IterableOnce[(Tensor[D1], Tensor[D2])]
class Object
trait Matchable
class Any
Show all
Generated with