Package io.github.kirstenali.deepj.nn

Class TensorJoin

java.lang.Object

io.github.kirstenali.deepj.nn.TensorJoin

public final class TensorJoin
extends Object

Backend copies plus a slicing autograd rule; never reads tensor values on the host.

Method Summary

Modifier and Type	Method	Description
static Tensor	concatenate(int axis, Tensor... inputs)
static Tensor	stack(int axis, Tensor... inputs)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

concatenate

public static Tensor concatenate(int axis,
 Tensor... inputs)

stack

public static Tensor stack(int axis,
 Tensor... inputs)