package torch

  1. Overview
  2. Docs
PyTorch bindings for OCaml

Install

Dune Dependency

Authors

Maintainers

Sources

0.5.tar.gz
md5=112bc4eef842883abdf25e150dc7952b
sha512=1670f337e5a00aa6acda9182b71c586cd817ed565ab1df280bba30f610760affb0d1fc859abe5f33bf78323cf3bf2391ad0b3d955aedff3894aae70723997705

doc/src/torch.vision/stb_image_write.ml.html

Source file stb_image_write.ml

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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
open Bigarray

type 'kind buffer = ('a, 'b, c_layout) Array1.t constraint 'kind = ('a, 'b) kind
type float32 = (float, float32_elt) kind
type int8 = (int, int8_unsigned_elt) kind

external png
  :  string
  -> w:int
  -> h:int
  -> c:int
  -> int8 buffer
  -> unit
  = "ml_stbi_write_png"

external bmp
  :  string
  -> w:int
  -> h:int
  -> c:int
  -> int8 buffer
  -> unit
  = "ml_stbi_write_bmp"

external tga
  :  string
  -> w:int
  -> h:int
  -> c:int
  -> int8 buffer
  -> unit
  = "ml_stbi_write_tga"

external hdr
  :  string
  -> w:int
  -> h:int
  -> c:int
  -> float32 buffer
  -> unit
  = "ml_stbi_write_hdr"

external jpg
  :  string
  -> w:int
  -> h:int
  -> c:int
  -> quality:int
  -> int8 buffer
  -> unit
  = "ml_stbi_write_jpg_bytecode" "ml_stbi_write_jpg_native"
OCaml

Innovation. Community. Security.