Shorthand for a horizontal linear concatenation.
(
*images: Iterable[Float[Tensor, "channel _ _"]],
align: Literal["start", "center", "end", "left", "right"] = "start",
gap: int = 8,
gap_color: Color = 1,
)
| 178 | |
| 179 | |
| 180 | def vcat( |
| 181 | *images: Iterable[Float[Tensor, "channel _ _"]], |
| 182 | align: Literal["start", "center", "end", "left", "right"] = "start", |
| 183 | gap: int = 8, |
| 184 | gap_color: Color = 1, |
| 185 | ): |
| 186 | """Shorthand for a horizontal linear concatenation.""" |
| 187 | return cat( |
| 188 | "vertical", |
| 189 | *images, |
| 190 | align={ |
| 191 | "start": "start", |
| 192 | "center": "center", |
| 193 | "end": "end", |
| 194 | "left": "start", |
| 195 | "right": "end", |
| 196 | }[align], |
| 197 | gap=gap, |
| 198 | gap_color=gap_color, |
| 199 | ) |
| 200 | |
| 201 | |
| 202 | def add_border( |