Searched defs:TensorDataset (Results 1 – 5 of 5) sorted by relevance
16 struct TensorDataset : public Dataset<TensorDataset, TensorExample> { struct18 explicit TensorDataset(const std::vector<Tensor>& tensors) in TensorDataset() function21 explicit TensorDataset(torch::Tensor tensor) : tensor(std::move(tensor)) {} in TensorDataset() function24 TensorExample get(size_t index) override { in get()29 std::optional<size_t> size() const override { in size()33 Tensor tensor;
193 class TensorDataset(Dataset[Tuple[Tensor, ...]]): class
64 class TensorDataset : public CachableSequence<Tensor> { class
4691 class TensorDataset(DatasetSource): class
51537 func TensorDataset(scope *Scope, components []tf.Output, output_shapes []tf.Shape, optional ...Tens… func