Docs.rs
tch-0.7.2
tch 0.7.2
Docs.rs crate page
MIT/Apache-2.0
Links
Repository
crates.io
Source
Owners
LaurentMazare
Dependencies
cpython ^0.2.0
normal
half ^1.6
normal
lazy_static ^1.3.0
normal
libc ^0.2.0
normal
ndarray ^0.15
normal
rand ^0.8
normal
thiserror ^1
normal
torch-sys ^0.7.2
normal
zip ^0.5
normal
anyhow ^1
dev
Versions
41.5%
of the crate is documented
Go to latest version
Platform
i686-unknown-linux-gnu
x86_64-unknown-linux-gnu
Feature flags
Rust
About docs.rs
Privacy policy
Rust website
The Book
Standard Library API Reference
Rust by Example
The Cargo Guide
Clippy Documentation
☰
Conv2D
Trait Implementations
Module
In tch::nn
?
Type Definition
tch
::
nn
::
Conv2D
source
·
[
−
]
pub type Conv2D =
Conv
<
[
i64
; 2]
>;
Expand description
Two dimensions convolution layer.
Trait Implementations
source
impl
Module
for
Conv2D
source
fn
forward
(&self, xs: &
Tensor
) ->
Tensor
