kshitij12345 0527d14248 [numpy] Add torch.take_along_dim (#52833) ... Summary: Reference: https://github.com/pytorch/pytorch/issues/38349 Wrapper around the existing `torch.gather` with broadcasting logic. TODO: * [x] Add Doc entry (see if phrasing can be improved) * [x] Add OpInfo * [x] Add test against numpy * [x] Handle broadcasting behaviour and when dim is not given. Pull Request resolved: https://github.com/pytorch/pytorch/pull/52833 Reviewed By: malfet Differential Revision: D27319038 Pulled By: mruberry fbshipit-source-id: 00f307825f92c679d96e264997aa5509172f5ed1		2021-03-28 05:22:51 -07:00
..
_static	Add documentation page for pipeline parallelism. (#50791)	2021-01-25 13:47:13 -08:00
_templates	various doc building cleanups (#53851)	2021-03-16 15:01:59 -07:00
community
notes	Fix inport -> import typo in documentation (#53589)	2021-03-08 23:56:42 -08:00
rpc	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
scripts
__config__.rst	Fix __config__ docs (#48557)	2020-11-29 23:57:06 -08:00
amp.rst
autograd.rst
backends.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
benchmark_utils.rst	Expand benchmark utils docs (#51664)	2021-02-04 00:22:41 -08:00
bottleneck.rst
checkpoint.rst
complex_numbers.rst	various doc building cleanups (#53851)	2021-03-16 15:01:59 -07:00
conf.py	various doc building cleanups (#53851)	2021-03-16 15:01:59 -07:00
cpp_extension.rst
cpp_index.rst
cuda.rst	expose memory_fraction and gpu_process docs (#51372)	2021-01-29 18:22:34 -08:00
cudnn_persistent_rnn.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
cudnn_rnn_determinism.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
data.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
ddp_comm_hooks.rst	[Gradient Compression] Introduce fp16_compress_wrapper in ddp_comm_hooks.rst (#54052)	2021-03-16 15:40:10 -07:00
distributed.optim.rst	[Reland] Update and expose ZeroRedundancyOptimizer docs (#53112)	2021-03-02 14:16:12 -08:00
distributed.rst	update distributed doc table for alltoall nccl (#54277)	2021-03-19 15:35:10 -07:00
distributions.rst	Add sample validation for LKJCholesky.log_prob (#52763)	2021-02-25 16:12:29 -08:00
dlpack.rst
docutils.conf
fft.rst	Remove deprecated spectral ops from torch namespace (#48594)	2020-12-05 04:12:32 -08:00
futures.rst
fx.rst	[FX][docs] Render inherited methods in fx.Tracer API reference (#53630)	2021-03-09 14:30:41 -08:00
hub.rst
index.rst	[special] add torch.special namespace (#52296)	2021-03-04 00:04:36 -08:00
jit.rst	Improve nn.Sequential documentation (#53380)	2021-03-24 13:02:43 -07:00
jit_builtin_functions.rst
jit_language_reference.rst	add type annotations to torch.nn.modules.conv (#49564)	2021-01-15 11:16:11 -08:00
jit_language_reference_v2.rst	[docs] Add starter content for new TorchScript language reference (#53837)	2021-03-11 18:18:27 -08:00
jit_python_reference.rst
jit_unsupported.rst
linalg.rst	Added torch.linalg.matrix_power (#52608)	2021-03-23 15:10:06 -07:00
math-quantizer-equation.png
mobile_optimizer.rst	Mod lists to neutral+descriptive terms in caffe2/docs (#49803)	2020-12-23 11:37:11 -08:00
model_zoo.rst
multiprocessing.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
name_inference.rst
named_tensor.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
nn.functional.rst	Huber loss (#50553)	2021-03-02 17:30:45 -08:00
nn.init.rst
nn.rst	Grammatically update tech docs (#54370)	2021-03-23 02:59:19 -07:00
onnx.rst	[ONNX] Update ONNX documentation (#51362) (#53313)	2021-03-12 02:49:38 -08:00
optim.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
pipeline.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
profiler.rst	Update docs for torch.profiler.tensorboard_trace_handler (#51636)	2021-02-04 13:32:59 -08:00
quantization-support.rst	[quantization] Add some support for 3d operations (#50003)	2021-03-10 16:40:35 -08:00
quantization.rst	Fix quantization doc issue (#50187)	2021-02-02 20:33:21 -08:00
random.rst
rpc.rst	Add 'remote_parameters' and 'get_module_rref' to RemoteModule docs. (#54645)	2021-03-26 21:41:28 -07:00
sparse.rst	[wip] doc_fix (#51825)	2021-02-06 11:36:36 -08:00
special.rst	[special] Add special.entr (#53500)	2021-03-24 18:44:42 -07:00
storage.rst	[docs] Add docs for storage and tensors for quantized Tensor (#51817)	2021-02-09 13:20:56 -08:00
tensor_attributes.rst	Fix the repeated entry in the Tensor Attributes doc (#52995)	2021-03-01 16:42:32 -08:00
tensor_view.rst	Add torch.swapdims and torch.swapaxes (#46041)	2020-11-18 11:35:53 -08:00
tensorboard.rst
tensors.rst	[numpy] Add torch.take_along_dim (#52833)	2021-03-28 05:22:51 -07:00
torch.nn.intrinsic.qat.rst	[quantization] Add some support for 3d operations (#50003)	2021-03-10 16:40:35 -08:00
torch.nn.intrinsic.quantized.rst
torch.nn.intrinsic.rst	[quantization] Add some support for 3d operations (#50003)	2021-03-10 16:40:35 -08:00
torch.nn.qat.rst
torch.nn.quantized.dynamic.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
torch.nn.quantized.rst	[quant] add docs for embedding/embedding_bag (#51770)	2021-02-05 11:43:15 -08:00
torch.overrides.rst	Add documentation for torch.overrides submodule. (#48170)	2020-11-30 11:25:31 -08:00
torch.quantization.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
torch.rst	[numpy] Add torch.take_along_dim (#52833)	2021-03-28 05:22:51 -07:00
type_info.rst