def chunk(self, n_chunks, dim=0):
"""Splits this tensor into a tuple of tensors.
See :func:`torch.chunk`.
"""
return torch.chunk(self, n_chunks, dim)
微信
公众号
扫码关注公众号