OpBMMExtensions Module

Type extensions

Type extension Description

this.bmm b

Full Usage: this.bmm b

Parameters:
Returns: Tensor

Extended Type: Tensor

b : Tensor
Returns: Tensor

dsharp.bmm (a, b)

Full Usage: dsharp.bmm (a, b)

Parameters:
    a : Tensor - The first tensor.
    b : Tensor - The second tensor.

Returns: Tensor

Batched matrix product of two tensors.

Extended Type: dsharp

a : Tensor

The first tensor.

b : Tensor

The second tensor.

Returns: Tensor

© Copyright 2021, DiffSharp Contributors.