pub fn sub(
a: &CudaTensor,
b: &CudaTensor,
stream: &CudaStream,
) -> CudaResult<CudaTensor>Expand description
Subtract two tensors
pub fn sub(
a: &CudaTensor,
b: &CudaTensor,
stream: &CudaStream,
) -> CudaResult<CudaTensor>Subtract two tensors