美文网首页
torch modules

torch modules

作者: DeepWeaver | 来源:发表于2017-12-06 21:05 被阅读3次
class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1,
padding=0, dilation=1, groups=1, bias=True)
torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)

Returns the maximum value of each row of the input tensor in the given dimension dim. The second return value is the index location of each maximum value found (argmax).

相关文章

网友评论

      本文标题:torch modules

      本文链接:https://www.haomeiwen.com/subject/xdnlixtx.html