Sdílet prostřednictvím


AutoFormerV2Block.forward(torch+Tensor, Int32, Int32, torch+Tensor) Method

Definition

TorchSharp.torch.nn.Module`5.forward(TorchSharp.torch.Tensor,System.Int32,System.Int32,TorchSharp.torch.Tensor)
public override TorchSharp.torch.Tensor forward (TorchSharp.torch.Tensor x, int h, int w, TorchSharp.torch.Tensor maskMatrix);
override this.forward : TorchSharp.torch.Tensor * int * int * TorchSharp.torch.Tensor -> TorchSharp.torch.Tensor
Public Overrides Function forward (x As torch.Tensor, h As Integer, w As Integer, maskMatrix As torch.Tensor) As torch.Tensor

Parameters

x
TorchSharp.torch.Tensor
h
Int32
w
Int32
maskMatrix
TorchSharp.torch.Tensor

Returns

TorchSharp.torch.Tensor

Applies to