-
Notifications
You must be signed in to change notification settings - Fork 89
Expand file tree
/
Copy pathtorch_dstack.Rd
More file actions
35 lines (30 loc) · 835 Bytes
/
torch_dstack.Rd
File metadata and controls
35 lines (30 loc) · 835 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/gen-namespace-docs.R,
% R/gen-namespace-examples.R, R/gen-namespace.R
\name{torch_dstack}
\alias{torch_dstack}
\title{Dstack}
\usage{
torch_dstack(tensors)
}
\arguments{
\item{tensors}{(sequence of Tensors) sequence of tensors to concatenate}
}
\description{
Dstack
}
\section{dstack(tensors, *, out=None) -> Tensor }{
Stack tensors in sequence depthwise (along third axis).
This is equivalent to concatenation along the third axis after 1-D and 2-D
tensors have been reshaped by \code{\link[=torch_atleast_3d]{torch_atleast_3d()}}.
}
\examples{
if (torch_is_installed()) {
a <- torch_tensor(c(1, 2, 3))
b <- torch_tensor(c(4, 5, 6))
torch_dstack(list(a,b))
a <- torch_tensor(rbind(1,2,3))
b <- torch_tensor(rbind(4,5,6))
torch_dstack(list(a,b))
}
}