burn-tensor 0.16.1

Tensor library with user-friendly APIs and automatic differentiation support
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
mod adaptive_avgpool1d;
mod adaptive_avgpool2d;
mod avgpool1d;
mod avgpool2d;
mod bicubic_interpolate;
mod bilinear_interpolate;
mod conv1d;
mod conv2d;
mod conv3d;
mod conv_transpose1d;
mod conv_transpose2d;
mod conv_transpose3d;
mod deform_conv2d;
mod forward;
mod maxpool1d;
mod maxpool2d;
mod nearest_interpolate;
mod unfold4d;