PyTorch 1.3.0 |
PyTorch, , 1.3.0 , .
:
NCHW = [N, C, H, W]
images = torch.randn(32, 3, 56, 56, names=NCHW)
images.sum('C')
images.select('C', index=0)
, Pytorch Developer Conference 2019.
http://feedproxy.google.com/~r/org/LOR/~3/rXeqG8rBULg/15281787