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
36
37
pub struct Scalar {
pub(crate) c_scalar: *mut torch_sys::C_scalar,
}
impl Scalar {
pub fn int(v: i64) -> Scalar {
let c_scalar = unsafe_torch!({ torch_sys::ats_int(v) });
Scalar { c_scalar }
}
pub fn float(v: f64) -> Scalar {
let c_scalar = unsafe_torch!({ torch_sys::ats_float(v) });
Scalar { c_scalar }
}
}
impl Drop for Scalar {
fn drop(&mut self) {
unsafe_torch!({ torch_sys::ats_free(self.c_scalar) })
}
}
impl From<i64> for Scalar {
fn from(v: i64) -> Scalar {
Scalar::int(v)
}
}
impl From<f64> for Scalar {
fn from(v: f64) -> Scalar {
Scalar::float(v)
}
}